experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

renaming namespace cuda_helper to gpu_helper.

Browse Source 
Also adding a `cuda_helper` as an alias for `gpu_helper` in non ROCm mode (for backwards compatibility)
This commit is contained in:
Deven Desai 2019-05-19 18:48:47 +00:00
parent e0c05bb427
commit 3e9beeb2f8
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/util/gpu_kernel_helper.h
View File

@ -160,7 +160,7 @@ __device__ EIGEN_ALWAYS_INLINE Eigen::half GpuShuffleXorSync(
// Aliased in gpu_device_functions.h // Aliased in gpu_device_functions.h
#endif #endif
namespace cuda_helper { namespace gpu_helper {
template <typename T, typename OutType = int32> template <typename T, typename OutType = int32>
__device__ OutType upper_bound(const T* first, OutType count, T val) { __device__ OutType upper_bound(const T* first, OutType count, T val) {
const T* orig = first; const T* orig = first;
@ -202,6 +202,11 @@ __device__ OutType lower_bound(const T* first, OutType count, T val) {
} }
} // namespace cuda_helper } // namespace cuda_helper
#ifndef TENSORFLOW_USE_ROCM
namespace cuda_helper = gpu_helper;
#endif
} // namespace tensorflow } // namespace tensorflow
#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM