SWDEV-381627 - Move cl_vk_amd.hpp into rocclr

Rename VK interop to ExternalMemory object, since it should handle DX interops also Change-Id: I536ec46d3e53ece35234a2e29030393ad411b96d [ROCm/clr commit: 3e5803c4c0]
2023-04-19 16:48:00 -04:00
parent a00b618051
commit b2525dfb82
6 changed files with 123 additions and 163 deletions
@@ -0,0 +1,95 @@
+/* Copyright (c) 2010 - 2021 Advanced Micro Devices, Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE. */
+
+#pragma once
+
+#include "platform/context.hpp"
+#include "platform/memory.hpp"
+
+namespace amd
+{
+  class ExternalMemory : public InteropObject
+  {
+  public:
+    enum class HandleType : uint32_t {
+      OpaqueFd = 1,
+      OpaqueWin32 = 2,
+      OpaqueWin32Kmt = 3,
+      D3D12Heap = 4,
+      D3D12Resource = 5,
+      D3D11Resource = 6,
+      D3D11ResourceKmt = 7
+    };
+
+    ExternalMemory(amd::Os::FileDesc handle, ExternalMemory::HandleType handle_type)
+        : handle_(handle), handle_type_(handle_type) {}
+
+    virtual ~ExternalMemory() override {}
+    ExternalMemory* asExternalMemory() final { return this; }
+
+    amd::Os::FileDesc Handle() const { return handle_; }
+    HandleType Type() const { return handle_type_; }
+
+  protected:
+    amd::Os::FileDesc handle_;
+    ExternalMemory::HandleType handle_type_;
+  };
+
+  class ExternalBuffer : public Buffer, public ExternalMemory
+  {
+  protected:
+    // Initializes device memory array, which is located after ExternalBuffer object in memory
+    void initDeviceMemory() {
+      deviceMemories_ =
+        reinterpret_cast<DeviceMemory*>(reinterpret_cast<char*>(this) + sizeof(ExternalBuffer));
+      memset(deviceMemories_, 0, context_().devices().size() * sizeof(DeviceMemory));
+    }
+
+  public:
+    ExternalBuffer(Context& amdContext, size_t size_in_bytes, amd::Os::FileDesc handle,
+        ExternalMemory::HandleType handle_type)
+        : Buffer(amdContext, 0, size_in_bytes), ExternalMemory(handle, handle_type) {
+      setInteropObj(this);
+    }
+
+    virtual ~ExternalBuffer() final {}
+  };
+
+  // to be modified once image requirments are known, for now, implement like buffer
+  class ExternalImage : public Buffer, public ExternalMemory
+  {
+  protected:
+    // Initializes device memory array, which is located after ExternalImage object in memory
+    void initDeviceMemory() {
+      deviceMemories_ =
+        reinterpret_cast<DeviceMemory*>(reinterpret_cast<char*>(this) + sizeof(ExternalImage));
+      memset(deviceMemories_, 0, context_().devices().size() * sizeof(DeviceMemory));
+    }
+
+  public:
+    ExternalImage(Context& amdContext, size_t size_in_bytes, amd::Os::FileDesc handle,
+        ExternalMemory::HandleType handle_type)
+        : Buffer(amdContext, 0, size_in_bytes), ExternalMemory(handle, handle_type) {
+      setInteropObj(this);
+    }
+
+    virtual ~ExternalImage() final {}
+  };
+}
@@ -26,8 +26,7 @@ namespace amd {
 //! Forward declarations of interop classes
 class GLObject;
 class BufferGL;
-class VkObject;
-class BufferVk;
+class ExternalMemory;


 #ifdef _WIN32
@@ -45,9 +44,7 @@ class InteropObject {
  // Static cast functions for interop objects
  virtual GLObject* asGLObject() { return nullptr; }
  virtual BufferGL* asBufferGL() { return nullptr; }
-
-  virtual VkObject* asVkObject() { return nullptr; }
-  virtual BufferVk* asBufferVk() { return nullptr; }
+  virtual ExternalMemory* asExternalMemory() { return nullptr; }

 #ifdef _WIN32
  virtual D3D10Object* asD3D10Object() { return nullptr; }