P4 to Git Change 1809454 by kjayapra@0_HIPWS_TRT2_ROCM on 2019/06/11 17:45:36

SWDEV-191936 - Implementation of hipMemcpyPeer & hipMemcpyPeerAsync + fix to check source P2PAgents for P2P. RocM Verified.

Affected files ...

... //depot/stg/opencl/drivers/opencl/api/hip/hip_peer.cpp#6 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/rocm/rocvirtual.cpp#76 edit

rocclr/runtime/device/rocm/rocvirtual.cpp

@@ -1266,7 +1266,15 @@ void VirtualGPU::submitCopyMemoryP2P(amd::CopyMemoryP2PCommand& cmd) {
       p2pAllowed = true;
       break;
     }
+
+    for (auto agent: srcDevMem->dev().p2pAgents()) {
+      if (agent.handle == dev().getBackendDevice().handle) {
+        p2pAllowed = true;
+        break;
+      }
+    }
   }
+
   // Synchronize source and destination memory
   device::Memory::SyncFlags syncFlags;
   syncFlags.skipEntire_ = cmd.isEntireMemory();