diff --git a/3rdparty/cmake/FindPytorch.cmake b/3rdparty/cmake/FindPytorch.cmake index 31babdca149..08bf18b001f 100644 --- a/3rdparty/cmake/FindPytorch.cmake +++ b/3rdparty/cmake/FindPytorch.cmake @@ -20,7 +20,7 @@ macro(translate_arch_string input output) string(REGEX REPLACE "([1-9])([0-9])-real" "\\1.\\2" version "${input}") elseif("${input}" MATCHES "([0-9]+)") string(REGEX REPLACE "([1-9])([0-9])" "\\1.\\2+PTX" version "${input}") - elseif(input STREQUAL "native") + elseif("${input}" STREQUAL "native") set(version "Auto") else() message(FATAL_ERROR "Invalid architecture string: ${input}")