From 4492dedf165b0379d2c1f918d0cb45cf5d89ff3b Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Sun, 28 Jul 2024 00:55:40 +0900 Subject: [PATCH] fix hip path detection --- modules/rocm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/rocm.py b/modules/rocm.py index dfd24c133..6ed9de794 100644 --- a/modules/rocm.py +++ b/modules/rocm.py @@ -129,7 +129,7 @@ if sys.platform == "win32": return os.path.join(hip_path, str(latest)) def get_version() -> str: # cannot just run hipconfig as it requires Perl installed on Windows. - return os.path.basename(path) + return os.path.basename(path) or os.path.basename(os.path.dirname(path)) def get_agents() -> List[Agent]: return [Agent(x.split(' ')[-1].strip()) for x in spawn("hipinfo").split("\n") if x.startswith('gcnArchName:')]