Switch to hipLaunchHostFunc for HIP graph (#667)

[ROCm/rccl commit: 2288e9ae80]
Этот коммит содержится в:
Wenkai Du
2022-12-15 10:16:46 -08:00
коммит произвёл GitHub
родитель 78fc48f1c8
Коммит d95a71b2ef
+1 -1
Просмотреть файл
@@ -258,7 +258,7 @@ ncclResult_t ncclStrongStreamLaunchHost(
}
ss->serialEventNeedsRecord = true;
#else
CUDACHECK(hipStreamAddCallback(ss->cudaStream, (hipStreamCallback_t)fn, arg, 0));
CUDACHECK(cudaLaunchHostFunc(ss->cudaStream, fn, arg));
#endif
return ncclSuccess;
}