Merge branch 'fix_grout_norm' into '24.04.1-fastpt'

fixed a bug regarding PTX commands See merge request !7

Merge branch 'fix_grout_norm' into '24.04.1-fastpt'
fixed a bug regarding PTX commands See merge request !7
ce0ebb14 · limm · e70ac2fd · 20eda944 · ce0ebb14
Commit ce0ebb14 authored Jul 03, 2025 by limm
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

apex/contrib/csrc/group_norm/group_norm_nhwc_bwd_one_pass_kernel.cuh ...b/csrc/group_norm/group_norm_nhwc_bwd_one_pass_kernel.cuh +2 -1

No files found.
--- a/apex/contrib/csrc/group_norm/group_norm_nhwc_bwd_one_pass_kernel.cuh
+++ b/apex/contrib/csrc/group_norm/group_norm_nhwc_bwd_one_pass_kernel.cuh
@@ -328,7 +328,8 @@ __global__ __launch_bounds__(THREADS_PER_BLOCK_)
  // Mark the completion of the threadblock.
  if( threadIdx.x == 0 ) {
-    asm volatile("red.release.gpu.global.add.s32 [%0], 1;" :: "l"(barrier));
+    int one = 1;
+    asm volatile("red.release.gpu.global.add.s32 [%0], 1;" :: "l"(barrier), "r"(one));
  }
  // Exit if that's not the last thread block.