"vscode:/vscode.git/clone" did not exist on "83c871fbe206c4bba382ce7780c114d9fb2ce2d5"
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
......@@ -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