fix bug when compile with native cuda arch

isl-org · Nov 18, 2024 · a6af079 · a6af079
1 parent 9d0cfc8
commit a6af079
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}")