[ROCm] Get rocm version from /opt/rocm/.info/version (#5815)

Previously we used to get ROCm version information from /opt/rocm/.info/version-dev file. This PR is to modify the code to get ROCm version from /opt/rocm/.info/version file instead to add compatibility with ROCm Centos9 docker images. cc: @jithunnair-amd Co-authored-by: Logan Adams <[email protected]>
microsoft · Aug 1, 2024 · f82d088 · f82d088
1 parent 324ee65
commit f82d088
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -219,7 +219,7 @@ def installed_rocm_version():
         ROCM_VERSION_DEV_RAW = ""
         if OpBuilder.is_rocm_pytorch():
             from torch.utils.cpp_extension import ROCM_HOME
-            rocm_ver_file = Path(ROCM_HOME).joinpath(".info/version-dev")
+            rocm_ver_file = Path(ROCM_HOME).joinpath(".info/version")
             if rocm_ver_file.is_file():
                 with open(rocm_ver_file, 'r') as file:
                     ROCM_VERSION_DEV_RAW = file.read()