From 7f906ced10ba4a071a01fdb60f55b8ece148fc63 Mon Sep 17 00:00:00 2001
From: Danylo Lytovchenko <danylo.lytovchenko@amd.com>
Date: Thu, 22 Jan 2026 17:26:19 +0100
Subject: [PATCH] Revert "rocr: SVMPrefetch to a particular numa node (#1063)"
 (#2792)

This reverts commit 0ba5a01baa367c58361170521ed8f24dfc4798fb.
---
 projects/rocr-runtime/CMakeLists.txt          |  9 +-----
 .../hsa-runtime/core/runtime/runtime.cpp      | 28 ++++---------------
 2 files changed, 6 insertions(+), 31 deletions(-)

diff --git a/projects/rocr-runtime/CMakeLists.txt b/projects/rocr-runtime/CMakeLists.txt
index 794beef3cb..44850dd7cf 100644
--- a/projects/rocr-runtime/CMakeLists.txt
+++ b/projects/rocr-runtime/CMakeLists.txt
@@ -60,13 +60,6 @@ if (NOT DEFINED BUILD_ROCR)
   set(BUILD_ROCR ON)
 endif()
 
-find_package(NUMA)
-if(NUMA_FOUND)
-  set(NUMA "${NUMA_LIBRARIES}")
-else()
-  find_library(NUMA NAMES numa REQUIRED)
-endif()
-
 function(add_rocm_subdir subdir subdir_assigns)
     message("add_rocm_subdir() -- " ${subdir})
     # message("  subdir_assigns before:" ${subdir_assigns} "EOM")
@@ -245,7 +238,7 @@ set(CPACK_DEBIAN_DEV_PACKAGE_DEPENDS "libdrm-amdgpu-dev | libdrm-dev, rocm-core,
 
 set(CPACK_DEBIAN_DEV_PACKAGE_RECOMMENDS "libdrm-amdgpu-dev")
 
-set(CPACK_DEBIAN_BINARY_PACKAGE_DEPENDS "libdrm-amdgpu-amdgpu1 | libdrm-amdgpu1, libnuma1, numactl, libelf1")
+set(CPACK_DEBIAN_BINARY_PACKAGE_DEPENDS "libdrm-amdgpu-amdgpu1 | libdrm-amdgpu1, libnuma1, libelf1")
 set(CPACK_DEBIAN_ASAN_PACKAGE_DEPENDS "libdrm-amdgpu-dev | libdrm-dev, rocm-core-asan, libdrm-amdgpu-amdgpu1 | libdrm-amdgpu1, libnuma1, libelf1")
 set(CPACK_DEBIAN_ASAN_PACKAGE_RECOMMENDS "libdrm-amdgpu-dev")
 
diff --git a/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/runtime.cpp b/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/runtime.cpp
index b2cae0833a..4fbf5aa145 100644
--- a/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/runtime.cpp
+++ b/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/runtime.cpp
@@ -56,7 +56,6 @@
 #include <sys/mman.h>
 #include <sys/socket.h>
 #include <sys/un.h>
-#include <numaif.h>      // for move_pages
 #else
 #define debug_warning(__VA_ARGS__)
 #endif
@@ -3355,29 +3354,12 @@ hsa_status_t Runtime::SvmPrefetch(void* ptr, size_t size, hsa_agent_t agent,
           op->dep_signals[op->remaining_deps], HSA_SIGNAL_CONDITION_EQ, 0, signal_handler, arg);
       return false;
     }
-    
-    Agent* dest = Runtime::runtime_singleton_->GetSVMPrefetchAgent(op->base, op->size);
 
-    if (dest == nullptr || dest->device_type() == Agent::kAmdGpuDevice) {
-      // Prefetch location is not valid for move_pages usecase.
-      HSA_SVM_ATTRIBUTE attrib;
-      attrib.type = HSA_SVM_ATTR_PREFETCH_LOC;
-      attrib.value = op->node_id;
-      HSAKMT_STATUS error = HSAKMT_CALL(hsaKmtSVMSetAttr(op->base, op->size, 1, &attrib));
-      assert(error == HSAKMT_STATUS_SUCCESS && "KFD Prefetch failed.");
-    } else {
-      // Migrate pages to the requested CPU NUMA node
-      void* base_ptr = op->base;
-      size_t num_pages = op->size / 4096;
-      std::vector<void*> pages(num_pages);
-      for (size_t i = 0; i < num_pages; ++i)
-          pages[i] = static_cast<uint8_t*>(base_ptr) + i * 4096;
-      std::vector<int> nodes(num_pages, op->node_id);
-      std::vector<int> status(num_pages, -1);
-
-      int ret = move_pages(0, num_pages, pages.data(), nodes.data(), status.data(), 0);
-      assert(ret == 0 && "move_pages failed");
-    }
+    HSA_SVM_ATTRIBUTE attrib;
+    attrib.type = HSA_SVM_ATTR_PREFETCH_LOC;
+    attrib.value = op->node_id;
+    HSAKMT_STATUS error = HSAKMT_CALL(hsaKmtSVMSetAttr(op->base, op->size, 1, &attrib));
+    assert(error == HSAKMT_STATUS_SUCCESS && "KFD Prefetch failed.");
 
     removePrefetchRanges(op);