[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