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

Revert changes to gfx1030 (#622)

Browse Source 
[ROCm/rccl commit: d9216af48b]
This commit is contained in:
Wenkai Du
2022-09-20 20:06:17 -07:00
committed by GitHub
parent 98609a7b92
commit 81c71aeb67
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/collectives/device/prims_simple.h
+8
View File
@@ -319,7 +319,11 @@ private:
}
barrier(); // This barrier has a counterpart in following loop
#if defined(__gfx1030__)
if (Send && (flags & RolePostSend) && index == 0) __threadfence_system();
#else
if ((MaxSend == 0 || MaxRecv == 0) && Send && (flags & RolePostSend) && index == 0) __threadfence_system();
#endif
__syncwarp();
postPeer<Recv, Send>();
offset += sliceSize;
@@ -339,7 +343,11 @@ private:
waitPeer<DirectRecv, DirectSend, Recv, Send, Src, Dst>(0, 0, 0, 0);
}
barrier(); // Has couterpart in preceding worker-only loop.
#if defined(__gfx1030__)
if (Send && (flags & RolePostSend) && sliceSize > 0 && index == 0) __threadfence_system();
#else
if ((MaxSend == 0 || MaxRecv == 0) && Send && (flags & RolePostSend) && sliceSize > 0 && index == 0) __threadfence_system();
#endif
__syncwarp();
postPeer<Recv, Send>();
offset += sliceSize;