fix launch bounds for cleanup

a640c63b · ashishfarmer · f2fcce58 · a640c63b
Commit a640c63b authored Jun 22, 2020 by ashishfarmer
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 2 deletions

csrc/multi_tensor_l2norm_kernel.cu csrc/multi_tensor_l2norm_kernel.cu +10 -2

No files found.
--- a/csrc/multi_tensor_l2norm_kernel.cu
+++ b/csrc/multi_tensor_l2norm_kernel.cu
@@ -194,7 +194,11 @@ struct MaxNormFunctor
 };


-__global__ void cleanup(
+__global__ void
+#ifdef __HIP_PLATFORM_HCC__
+__launch_bounds__(1024)
+#endif
+cleanup(
  float* output,
  float* output_per_tensor,
  float* ret,
@@ -231,7 +235,11 @@ __global__ void cleanup(
  }
 }

-__global__ void cleanup_v2(
+__global__ void
+#ifdef __HIP_PLATFORM_HCC__
+__launch_bounds__(1024)
+#endif
+cleanup_v2(
  float* output,
  float* output_per_tensor,
  float* ret,