From 62aee13f7b4d6fd66efe979981b63f810e55e1d7 Mon Sep 17 00:00:00 2001
From: Flora Cui <flora.cui@amd.com>
Date: Sat, 5 Jul 2025 08:49:47 +0800
Subject: [PATCH] wsl/libhsakmt: fix hsaKmtRegisterGraphicsHandleToNodesExt

if nodes array is nullptr

Signed-off-by: Flora Cui <flora.cui@amd.com>
Reviewed-by: Tianci Yin <tianci.yin@amd.com>
Part-of: <http://10.67.69.192/wsl/rocr-runtime/-/merge_requests/83>
---
 memory.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/memory.cpp b/memory.cpp
index 13c0a82c3f..75bc01f331 100644
--- a/memory.cpp
+++ b/memory.cpp
@@ -451,6 +451,11 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtRegisterGraphicsHandleToNodesExt(HSAuint64 Graphic
   HSAKMT_STATUS ret = HSAKMT_STATUS_SUCCESS;
 
   pr_debug("number of nodes %lu\n", NumberOfNodes);
+  if (NumberOfNodes == 0) {
+    RegisterFlags.ui32.requiresVAddr = 0;
+    NumberOfNodes = 1;
+    NodeArray = (HSAuint32*)&(dxg_runtime->default_node);
+  }
 
   GraphicsResourceInfo->NodeId = NodeArray[0];
   return hsaKmtImportDMABufHandle(GraphicsResourceHandle,