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;