File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -325,8 +325,8 @@ if (onnxruntime_USE_ROCM)
325
325
# replicate strategy used by pytorch to get ROCM_VERSION
326
326
# https://github.com/pytorch/pytorch/blob/5c5b71b6eebae76d744261715231093e62f0d090/cmake/public/LoadHIP.cmake
327
327
# with modification
328
- if (EXISTS "${onnxruntime_ROCM_HOME} /.info/version-dev " )
329
- file (READ "${onnxruntime_ROCM_HOME} /.info/version-dev " ROCM_VERSION_DEV_RAW)
328
+ if (EXISTS "${onnxruntime_ROCM_HOME} /.info/version" )
329
+ file (READ "${onnxruntime_ROCM_HOME} /.info/version" ROCM_VERSION_DEV_RAW)
330
330
string (REGEX MATCH "^([0-9]+)\. ([0-9]+)\. ([0-9]+)-.*$" ROCM_VERSION_MATCH ${ROCM_VERSION_DEV_RAW} )
331
331
elseif (EXISTS "${onnxruntime_ROCM_HOME} /include/rocm_version.h" )
332
332
file (READ "${onnxruntime_ROCM_HOME} /include/rocm_version.h" ROCM_VERSION_H_RAW)
@@ -345,7 +345,7 @@ if (onnxruntime_USE_ROCM)
345
345
else ()
346
346
message (FATAL_ERROR "Cannot determine ROCm version string" )
347
347
endif ()
348
- message ("\n ***** ROCm version from ${onnxruntime_ROCM_HOME} /.info/version-dev ****\n " )
348
+ message ("\n ***** ROCm version from ${onnxruntime_ROCM_HOME} /.info/version ****\n " )
349
349
message ("ROCM_VERSION_DEV: ${ROCM_VERSION_DEV} " )
350
350
message ("ROCM_VERSION_DEV_MAJOR: ${ROCM_VERSION_DEV_MAJOR} " )
351
351
message ("ROCM_VERSION_DEV_MINOR: ${ROCM_VERSION_DEV_MINOR} " )
You can’t perform that action at this time.
0 commit comments