P4 to Git Change 1788476 by kjayapra@1_HIPWS_TRT2_PAL on 2019/05/28 14:44:38

SWDEV-190190 - Implemented API hipCanAccessPeerDevice

Affected files ...

... //depot/stg/opencl/drivers/opencl/api/hip/hip_peer.cpp#5 edit

hipamd/api/hip/hip_peer.cpp

@@ -53,7 +53,31 @@ hipError_t hipMemcpyPeerAsync(void* dst, hipCtx_t dstDevice, const void* src, hi
 hipError_t hipDeviceCanAccessPeer(int* canAccessPeer, int deviceId, int peerDeviceId) {
   HIP_INIT_API(canAccessPeer, deviceId, peerDeviceId);
 
-  *canAccessPeer = 0;
+  amd::Device* device = nullptr;
+  amd::Device* peer_device = nullptr;
+
+  if (canAccessPeer == nullptr) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
+
+  /* Peer cannot be self */
+  if (deviceId == peerDeviceId) {
+    *canAccessPeer = 0;
+    return HIP_RETURN(hipSuccess);
+  }
+
+  /* Cannot exceed the max number of devices */
+  if (static_cast<size_t>(deviceId) >= g_devices.size()
+       || static_cast<size_t>(peerDeviceId) >= g_devices.size()) {
+    return HIP_RETURN(hipErrorInvalidValue);
+  }
+
+  device = g_devices[deviceId]->devices()[0];
+  peer_device = g_devices[peerDeviceId]->devices()[0];
+
+  *canAccessPeer = static_cast<int>(std::find(device->p2pDevices_.begin(),
+                                              device->p2pDevices_.end(), as_cl(peer_device))
+                                              != device->p2pDevices_.end());
 
   return HIP_RETURN(hipSuccess);
 }
@@ -61,17 +85,13 @@ hipError_t hipDeviceCanAccessPeer(int* canAccessPeer, int deviceId, int peerDevi
 hipError_t hipDeviceDisablePeerAccess(int peerDeviceId) {
   HIP_INIT_API(peerDeviceId);
 
-  assert(0 && "Unimplemented");
+  HIP_RETURN(hipSuccess);
-
-  HIP_RETURN(hipErrorUnknown);
 }
 
 hipError_t hipDeviceEnablePeerAccess(int peerDeviceId, unsigned int flags) {
   HIP_INIT_API(peerDeviceId, flags);
 
-  assert(0 && "Unimplemented");
+  HIP_RETURN(hipSuccess);
-
-  HIP_RETURN(hipErrorUnknown);
 }
 
 hipError_t hipMemcpyPeer(void* dst, int dstDevice, const void* src, int srcDevice,
@@ -95,15 +115,11 @@ hipError_t hipMemcpyPeerAsync(void* dst, int dstDevice, const void* src, int src
 hipError_t hipCtxEnablePeerAccess(hipCtx_t peerCtx, unsigned int flags) {
   HIP_INIT_API(peerCtx, flags);
 
-  assert(0 && "Unimplemented");
+  HIP_RETURN(hipSuccess);
-
-  HIP_RETURN(hipErrorUnknown);
 }
 
 hipError_t hipCtxDisablePeerAccess(hipCtx_t peerCtx) {
   HIP_INIT_API(peerCtx);
 
-  assert(0 && "Unimplemented");
+  HIP_RETURN(hipSuccess);
-
-  HIP_RETURN(hipErrorUnknown);
 }