Use hipGraph instead of cudaGraph (#487)

This commit is contained in:
Wenkai Du
2022-01-10 08:26:01 -08:00
committed by GitHub
szülő 565fbeb5e9
commit 3669e12432
4 fájl változott, egészen pontosan 6 új sor hozzáadva és 8 régi sor törölve
+1 -1
Fájl megtekintése
@@ -174,7 +174,7 @@ ncclResult_t ncclGroupEnd() {
for (int i=0; i<ncclGroupIndex; i++) doneArray[i] = 1;
ncclResult_t ret = ncclGroupError;
int usingCudaGraphAll = -1;
cudaGraph_t* graphs = NULL;
hipGraph_t* graphs = NULL;
if (ret != ncclSuccess) goto group_cleanup;
/* Launch async ncclCommInitRank */