From 3f66ce03f8c03b2e545f17407fe2eba622557b75 Mon Sep 17 00:00:00 2001
From: Flora Cui <flora.cui@amd.com>
Date: Thu, 17 Oct 2024 11:27:27 +0800
Subject: [PATCH] wsl/hsakmt: implement hsaKmtRegisterGraphicsHandleToNodesExt

Signed-off-by: Flora Cui <flora.cui@amd.com>
---
 memory.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/memory.cpp b/memory.cpp
index ce8fdbbf20..814df1c42b 100644
--- a/memory.cpp
+++ b/memory.cpp
@@ -268,6 +268,21 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtRegisterGraphicsHandleToNodes(HSAuint64 GraphicsRe
                                                             HsaGraphicsResourceInfo *GraphicsResourceInfo,
                                                             HSAuint64 NumberOfNodes,
                                                             HSAuint32 *NodeArray) {
+  HSA_REGISTER_MEM_FLAGS regFlags;
+  regFlags.Value = 0;
+
+  return hsaKmtRegisterGraphicsHandleToNodesExt(GraphicsResourceHandle,
+            GraphicsResourceInfo,
+            NumberOfNodes,
+            NodeArray,
+            regFlags);
+}
+
+HSAKMT_STATUS HSAKMTAPI hsaKmtRegisterGraphicsHandleToNodesExt(HSAuint64 GraphicsResourceHandle,
+							       HsaGraphicsResourceInfo *GraphicsResourceInfo,
+							       HSAuint64 NumberOfNodes,
+							       HSAuint32 *NodeArray,
+							       HSA_REGISTER_MEM_FLAGS RegisterFlags) {
   CHECK_DXG_OPEN();
   uint32_t *gpu_id_array = NULL;
   HSAKMT_STATUS ret = HSAKMT_STATUS_SUCCESS;