From 0de373288472d7085e85d24a15e2c2a85d986175 Mon Sep 17 00:00:00 2001
From: Seunghoon Lee <lshqqytiger@naver.com>
Date: Sun, 28 Sep 2025 11:31:43 +0900
Subject: [PATCH] prioritize python pacakage check over PATH

---
 modules/rocm.py | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/modules/rocm.py b/modules/rocm.py
index 8dfcd9ec8..3ffff069a 100644
--- a/modules/rocm.py
+++ b/modules/rocm.py
@@ -130,17 +130,16 @@ class Agent:
 
 
 def find() -> Union[Environment, None]:
+    try: # TheRock
+        import _rocm_sdk_core # pylint: disable=unused-import
+        return PythonPackageEnvironment()
+    except ImportError:
+        pass
+
+    # system-wide installation
     hip_path = shutil.which("hipconfig")
     if hip_path is not None:
-        py_path = os.path.dirname(sys.executable)
-        if hip_path.startswith(py_path):
-            try:
-                import _rocm_sdk_core # pylint: disable=unused-import
-                return PythonPackageEnvironment()
-            except ImportError:
-                pass
-        else:
-            return ROCmEnvironment(dirname(resolve_link(hip_path), 2))
+        return ROCmEnvironment(dirname(resolve_link(hip_path), 2))
 
     if sys.platform == "win32":
         hip_path = os.environ.get("HIP_PATH", None)