Merge pull request #23 from ashishfarmer/launch_bounds_fix

Fix launch bounds for cleanup(...) call

Merge pull request #23 from ashishfarmer/launch_bounds_fix
Fix launch bounds for cleanup(...) call
7e099371 · Peng · GitHub · f2fcce58 · a640c63b · 7e099371
Unverified Commit 7e099371 authored Jun 22, 2020 by Peng Committed by GitHub Jun 22, 2020
Show 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,