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

resolved conflicts, fixed the localNetCount/0 bug

Browse Source 
[ROCm/rccl commit: 0d515f9388]
This commit is contained in:
PedramAlizadeh
2023-12-18 08:11:34 +00:00
parent 9cb0f98e73 90d7d1b07b
commit 401a104c5a
12 changed files with 107 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/init.cc
+1 -1
View File
@@ -672,7 +672,7 @@ static ncclResult_t devCommSetup(ncclComm_t comm) {
NCCLCHECKGOTO(ncclCudaMemcpyAsync(devCommAndChans, &tmpCommAndChans, 1, comm->sharedRes->deviceStream.cudaStream), ret, fail);
exit:
CUDACHECK(cudaStreamSynchronize(comm->sharedRes->deviceStream.cudaStream));
NCCLCHECK(ncclStrongStreamSynchronize(&comm->sharedRes->deviceStream));
NCCLCHECK(ncclStrongStreamRelease(ncclCudaGraphNone(), &comm->sharedRes->deviceStream));
return ret;
fail: