Skip to content

Commit

Permalink
Removing definition of GPU_LAUNCHLK_KERNEL
Browse files Browse the repository at this point in the history
  • Loading branch information
jerryyin committed May 7, 2019
1 parent 47b374e commit 10803b7
Showing 1 changed file with 0 additions and 12 deletions.
12 changes: 0 additions & 12 deletions tensorflow/core/util/gpu_kernel_helper.h
Original file line number Diff line number Diff line change
Expand Up @@ -33,18 +33,6 @@ limitations under the License.
#define TF_RED_WARPSIZE 64
#endif

#if GOOGLE_CUDA
#define GPU_LAUNCH_KERNEL(kernel, block_count, threads_per_block, shared_mem, \
stream, ...) \
TF_CHECK_OK(CudaLaunchKernel(kernel, block_count, threads_per_block, \
shared_mem, stream, __VA_ARGS__));
#elif TENSORFLOW_USE_ROCM
#define GPU_LAUNCH_KERNEL(kernel, block_count, threads_per_block, shared_mem, \
stream, ...) \
hipLaunchKernelGGL(kernel, block_count, threads_per_block, shared_mem, \
stream, __VA_ARGS__);
#endif

// Deprecated, use 'for(int i : GpuGridRangeX(n))' instead.
#define GPU_1D_KERNEL_LOOP(i, n) \
for (int i : ::tensorflow::GpuGridRangeX<int>(n))
Expand Down

0 comments on commit 10803b7

Please sign in to comment.