[Openmp-commits] [openmp] [Libomptarget] Fix Nvidia offloading hanging on dataRetrieve using RPC (PR #66817)
Johannes Doerfert via Openmp-commits
openmp-commits at lists.llvm.org
Sat Sep 23 07:50:19 PDT 2023
================
@@ -576,7 +576,16 @@ struct CUDADeviceTy : public GenericDeviceTy {
if (auto Err = getStream(AsyncInfoWrapper, Stream))
return Err;
- CUresult Res = cuMemcpyDtoHAsync(HstPtr, (CUdeviceptr)TgtPtr, Size, Stream);
+ // If there is already pending work on the stream it could be waiting for
+ // someone to check the RPC server.
+ CUresult Res = cuStreamQuery(Stream);
----------------
jdoerfert wrote:
`if (auto RCPServer = getRPCServer())` on the outside, not in the while.
https://github.com/llvm/llvm-project/pull/66817
More information about the Openmp-commits
mailing list