prioritize python pacakage check over PATH

This commit is contained in:
Seunghoon Lee
2025-09-28 11:31:43 +09:00
parent 587db51e9e
commit 0de3732884
+8 -9
View File
@@ -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)