diff --git a/op_builder/builder.py b/op_builder/builder.py index b1ca311d0355..8998fc0eddb8 100644 --- 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()