Fixed issue with attomicEXCH creating errors on multi node runs (#587)

Этот коммит содержится в:
akolliasAMD
2022-07-22 11:32:49 -06:00
коммит произвёл GitHub
родитель 686dbc8bc6
Коммит 254208e7dd
+1 -1
Просмотреть файл
@@ -117,7 +117,7 @@ private:
if (flags & (Recv*RoleWaitRecv | Send*RoleWaitSend)) {
if (isSendNotRecv && (flags & SizesFifoEnabled))
atomicExch_system((unsigned long long *)(connSizesFifoPtr+step%NCCL_STEPS), nelts*sizeof(T));
__atomic_store_n((connSizesFifoPtr+step%NCCL_STEPS), nelts*sizeof(T), __ATOMIC_SEQ_CST);
void **ptrs = isSendNotRecv ? (ncclShmem->groups[group].dsts + Dst)
: (ncclShmem->groups[group].srcs + Src);