Fix unit test HIP graph error (#712)

This commit is contained in:
Wenkai Du
2023-03-20 15:34:09 -07:00
committed by GitHub
orang tua 8c56e6c892
melakukan b02fd04165
+1 -1
Melihat File
@@ -938,7 +938,7 @@ ncclResult_t ncclLaunchPrepare(struct ncclComm* comm) {
NCCLCHECKGOTO(ncclStrongStreamWaitStream(tasks->capturingGraph, &comm->deviceStream, l->stream), result, failure);
}
NCCLCHECKGOTO(ncclStrongStreamWaitStream(tasks->capturingGraph, launchStream, &comm->deviceStream), result, failure);
} else if (tasks->streams->stream != comm->lastStream) {
} else if (tasks->streams->stream != comm->lastStream && comm->lastStream != nullptr) {
// Stream changed from last call, create dependency against last NCCL kernel launch
CUDACHECK(hipStreamWaitEvent(tasks->streams->stream, comm->doneEvent, 0));
}