* Revert "Ensure memory copy integrity during transport setup (#731)"
This reverts commit 36e453c61e.
Add stream synchronization in ncclStrongStreamRelease.
* Use event record and wait
This commit is contained in:
1 bovenliggende
c88475462b
commit
4ca7742c61
2 files changed
+4
-15
No files matched your search
@@ -237,6 +237,8 @@ ncclResult_t ncclStrongStreamRelease(struct ncclCudaGraph graph, struct ncclStro
|
||||
}
|
||||
}
|
||||
#endif
|
||||
CUDACHECK(cudaEventRecord(ss->scratchEvent, ss->cudaStream));
|
||||
CUDACHECK(cudaStreamWaitEvent(ss->cudaStream, ss->scratchEvent, 0));
|
||||
return ncclSuccess;
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user