Switch to hipLaunchHostFunc for HIP graph (#667)
Este cometimento está contido em:
cometido por
GitHub
ascendente
24aa8bd802
cometimento
2288e9ae80
@@ -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;
|
||||
}
|
||||
|
||||
Criar uma nova questão referindo esta
Bloquear um utilizador