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

* Revert "Ensure memory copy integrity during transport setup (#731)" This reverts commit 0c0c35927a. Add stream synchronization in ncclStrongStreamRelease. * Use event record and wait [ROCm/rccl commit: 4ca7742c61]
2023-05-16 10:34:47 -07:00
parent a26ca6d766
commit f7184aef3f
2 changed files with 4 additions and 15 deletions
@@ -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;
 }