Merge remote-tracking branch 'origin/master' into v2.19

Этот коммит содержится в:
Sylvain Jeaugey
2023-09-26 05:55:56 -07:00
родитель f9c3dc251e 4365458757
Коммит 3435178b6c
5 изменённых файлов: 21 добавлений и 13 удалений
+1 -1
Просмотреть файл
@@ -466,7 +466,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: