Skip to content

GitLab

Unverified Commit 40a6f53f authored by Inoki's avatar Inoki Committed by GitHub
Browse files

Display warning only when ROCm version is less than Pytorch required version (#29200) 


Signed-off-by: default avatarInoki <inoki@inoki.cc>
parent ce58fdc1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
CMakeLists.txt
View file @ 40a6f53f
......@@ -136,7 +136,7 @@ elseif(HIP_FOUND)
# ROCm 5.X and 6.X
if (ROCM_VERSION_DEV_MAJOR GREATER_EQUAL 5 AND
NOT Torch_VERSION VERSION_EQUAL ${TORCH_SUPPORTED_VERSION_ROCM})
Torch_VERSION VERSION_LESS ${TORCH_SUPPORTED_VERSION_ROCM})
message(WARNING "Pytorch version >= ${TORCH_SUPPORTED_VERSION_ROCM} "
"expected for ROCm build, saw ${Torch_VERSION} instead.")
endif()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment