Rework barrier and event code (#761)

* Rework barrier and event code

* Switch to inline asm
Cette révision appartient à :
Wenkai Du a créé et GitHub a révisé 2023-05-31 13:36:51 -07:00
1 Parent 7519ecb476
révision 5a38ff192b
3 fichiers modifiés
+5 -6

Aucun fichier ne correspond à votre recherche

-2
Voir le fichier
@@ -237,8 +237,6 @@ ncclResult_t ncclStrongStreamRelease(struct ncclCudaGraph graph, struct ncclStro
}
}
#endif
CUDACHECK(cudaEventRecord(ss->scratchEvent, ss->cudaStream));
CUDACHECK(cudaStreamWaitEvent(ss->cudaStream, ss->scratchEvent, 0));
return ncclSuccess;
}