Switch to hipLaunchHostFunc for HIP graph (#667)
[ROCm/rccl commit: 2288e9ae80]
This commit is contained in:
1 orang tua
78fc48f1c8
melakukan
d95a71b2ef
1 file changed
+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;
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user