msccl: fix data corruption with MTYPE_RW (#1014)

[ROCm/rccl commit: 81602814a7]
Αυτή η υποβολή περιλαμβάνεται σε:
Wenkai Du
2023-12-11 20:33:15 -08:00
υποβλήθηκε από GitHub
γονέας fe5a902f97
υποβολή 48107b18c9
@@ -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++;
}
}