AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add back __syncthreads() in barrier and adjust stack size (#688)

Browse Source 
[ROCm/rccl commit: 1c166046a2]
This commit is contained in:
Wenkai Du
2023-02-18 08:50:31 -08:00
committed by GitHub
parent 7c1290f995
commit 393d0ba7f8
4 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/init.cc
+1 -1
View File
@@ -1415,7 +1415,7 @@ fail:
#ifdef USE_INDIRECT_FUNCTION_CALL
NCCL_PARAM(SetStackSize, "SET_STACK_SIZE", 1);
RCCL_PARAM(StackSizeOverride, "STACK_SIZE_OVERRIDE", 8);
RCCL_PARAM(StackSizeOverride, "STACK_SIZE_OVERRIDE", 256);
#else
NCCL_PARAM(SetStackSize, "SET_STACK_SIZE", 0);
RCCL_PARAM(StackSizeOverride, "STACK_SIZE_OVERRIDE", 0);