AI/rocm-systems
2
0
Forka 0
Codice Problemi Pull Request Actions Pacchetti Progetti Rilasci Wiki Attività

msccl: fix data corruption with MTYPE_RW (#1014)

Sfoglia il codice sorgente 
[ROCm/rccl commit: 81602814a7]
This commit is contained in:
Wenkai Du
2023-12-11 20:33:15 -08:00
committato da GitHub
parent fe5a902f97
commit 48107b18c9
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
Scarica il file Patch Scarica il file Diff Expand all files Collapse all files
projects/rccl/src/collectives/device/msccl_kernel_impl.h
+1 -1
Vedi File
@@ -460,7 +460,7 @@ __device__ __forceinline__ void mscclRunInterpreter(
return;
}
if (t->hasDependence && tid == nthreads-1)
__atomic_store_n(&mscclFlags[bid].flag, (uint64_t) COMPUTE_FLAG(workIndex, iter, step), __ATOMIC_RELAXED);
__atomic_store_n(&mscclFlags[bid].flag, (uint64_t) COMPUTE_FLAG(workIndex, iter, step), t->type == MSCCL_REDUCE ? __ATOMIC_RELEASE : __ATOMIC_RELAXED);
step++;
}
}