update rocm_version

48e6bbba · panning · cdc05670 · 48e6bbba
Commit 48e6bbba authored Jun 08, 2023 by panning
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -64,7 +64,7 @@ if sha != 'Unknown':
 dcu_version += "." + get_abi()
 if os.getenv("ROCM_PATH"):
    rocm_path = os.getenv('ROCM_PATH', "")
-    rocm_version_path = os.path.join(rocm_path, '.info', "version-dev")
+    rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version")
    with open(rocm_version_path, 'r',encoding='utf-8') as file:
        lines = file.readlines()
    rocm_version=lines[0][:-2].replace(".", "")