[DCU] little fix

73f3ac47 · yuguo · 456a96c8 · 73f3ac47
Commit 73f3ac47 authored Apr 22, 2025 by yuguo
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

transformer_engine/common/comm_gemm_overlap/userbuffers/userbuffers.cu ...ngine/common/comm_gemm_overlap/userbuffers/userbuffers.cu +1 -1

No files found.
--- a/transformer_engine/common/comm_gemm_overlap/userbuffers/userbuffers.cu
+++ b/transformer_engine/common/comm_gemm_overlap/userbuffers/userbuffers.cu
@@ -2157,7 +2157,7 @@ __global__ void __launch_bounds__(MAX_THREADS)
    atomicAdd_system(flagptr,
                     1);  // otherwise need local SM sync before sending flag
  } else {                // 0 bytes and 1 SM only
-#ifdef defined(__gfx928__) || defined(__gfx926__) || defined(__gfx906__)   
+#if defined(__gfx928__) || defined(__gfx926__) || defined(__gfx906__)   
    *flagptr = *flagptr + 1;
 #else
    atomicAdd_system(flagptr, 1);