P4 to Git Change 1736033 by kjayapra@9_HIPWS_IPCCHKIN on 2019/01/25 17:44:48

SWDEV-145570 - IPC Mem Handle Changes for HIP.

Affected files ...

... //depot/stg/opencl/drivers/opencl/api/hip/hip_internal.hpp#20 edit
... //depot/stg/opencl/drivers/opencl/api/hip/hip_memory.cpp#43 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/device.hpp#330 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/rocm/rocdevice.cpp#112 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/rocm/rocdevice.hpp#33 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/rocm/rocmemory.cpp#42 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/rocm/rocmemory.hpp#13 edit
... //depot/stg/opencl/drivers/opencl/runtime/platform/memory.hpp#108 edit

rocclr/runtime/device/rocm/rocdevice.cpp

@@ -1570,6 +1570,59 @@ void Device::updateFreeMemory(size_t size, bool free) {
   }
 }
 
+amd::Memory *Device::IpcAttach(const void* handle, size_t mem_size, unsigned int flags, void** dev_ptr) const {
+  amd::Memory* amd_mem_obj = nullptr;
+  hsa_status_t hsa_status = HSA_STATUS_SUCCESS;
+
+  /* Retrieve the devPtr from the handle */
+  hsa_agent_t hsa_agent = getBackendDevice();
+  hsa_status
+    = hsa_amd_ipc_memory_attach(reinterpret_cast<const hsa_amd_ipc_memory_t*>(handle),
+                                mem_size, 1, &hsa_agent, dev_ptr);
+
+  if (hsa_status != HSA_STATUS_SUCCESS) {
+    LogError("[OCL] HSA failed to attach IPC memory");
+    return nullptr;
+  }
+
+  /* Create an amd Memory object for the pointer */
+  amd_mem_obj = new (context()) amd::Buffer(context(), flags, mem_size, *dev_ptr);
+  if (amd_mem_obj == nullptr) {
+    LogError("[OCL] failed to create a mem object!");
+    return nullptr;
+  }
+
+  if (!amd_mem_obj->create(nullptr)) {
+    LogError("[OCL] failed to create a svm hidden buffer!");
+    amd_mem_obj->release();
+    return nullptr;
+  }
+
+  return amd_mem_obj;
+}
+
+void Device::IpcDetach (amd::Memory& memory) const {
+  void* dev_ptr = nullptr;
+  hsa_status_t hsa_status = HSA_STATUS_SUCCESS;
+
+  if(memory.getSvmPtr() != nullptr) {
+    dev_ptr = memory.getSvmPtr();
+  } else if (memory.getHostMem() != nullptr) {
+    dev_ptr = memory.getHostMem();
+  } else {
+    ShouldNotReachHere();
+  }
+
+  /*Detach the memory from HSA */
+  hsa_status = hsa_amd_ipc_memory_detach(dev_ptr);
+  if (hsa_status != HSA_STATUS_SUCCESS) {
+    LogError("[OCL] HSA failed to detach memory !");
+    return;
+  }
+
+  memory.release();
+}
+
 void* Device::svmAlloc(amd::Context& context, size_t size, size_t alignment, cl_svm_mem_flags flags,
                        void* svmPtr) const {
   amd::Memory* mem = nullptr;