Switch to hipLaunchHostFunc for HIP graph (#667)
[ROCm/rccl commit: 2288e9ae80]
This commit is contained in:
gecommit door
GitHub
bovenliggende
78fc48f1c8
commit
d95a71b2ef
@@ -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;
|
||||
}
|
||||
|
||||
Verwijs in nieuw issue
Block a user