Skip to content

Commit 40a6f53

Browse files
authored
Display warning only when ROCm version is less than Pytorch required version (#29200)
Signed-off-by: Inoki <inoki@inoki.cc>
1 parent ce58fdc commit 40a6f53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ elseif(HIP_FOUND)
136136

137137
# ROCm 5.X and 6.X
138138
if (ROCM_VERSION_DEV_MAJOR GREATER_EQUAL 5 AND
139-
NOT Torch_VERSION VERSION_EQUAL ${TORCH_SUPPORTED_VERSION_ROCM})
139+
Torch_VERSION VERSION_LESS ${TORCH_SUPPORTED_VERSION_ROCM})
140140
message(WARNING "Pytorch version >= ${TORCH_SUPPORTED_VERSION_ROCM} "
141141
"expected for ROCm build, saw ${Torch_VERSION} instead.")
142142
endif()

0 commit comments

Comments
 (0)