Revert "Ensure memory copy integrity during transport setup (#731)" (#741)

* 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:
Wenkai Du authored and GitHub committed 2023-05-16 10:34:47 -07:00
1 bovenliggende c88475462b
commit 4ca7742c61
2 files changed
+4 -15

No files matched your search

+2
Bestand weergeven
@@ -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;
}