Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "d793ca17ac3b1206f420f273c1747e648cde96f6"
Commit 73f3ac47 authored by yuguo's avatar yuguo
Browse files

[DCU] little fix

parent 456a96c8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
transformer_engine/common/comm_gemm_overlap/userbuffers/userbuffers.cu
View file @ 73f3ac47
...@@ -2157,7 +2157,7 @@ __global__ void __launch_bounds__(MAX_THREADS) ...@@ -2157,7 +2157,7 @@ __global__ void __launch_bounds__(MAX_THREADS)
atomicAdd_system(flagptr, atomicAdd_system(flagptr,
1); // otherwise need local SM sync before sending flag 1); // otherwise need local SM sync before sending flag
} else { // 0 bytes and 1 SM only } else { // 0 bytes and 1 SM only
#ifdef defined(__gfx928__) || defined(__gfx926__) || defined(__gfx906__) #if defined(__gfx928__) || defined(__gfx926__) || defined(__gfx906__)
*flagptr = *flagptr + 1; *flagptr = *flagptr + 1;
#else #else
atomicAdd_system(flagptr, 1); atomicAdd_system(flagptr, 1);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment