fix bug when compile with native cuda arch (#7059)

isl-org · Nov 22, 2024 · 11aa48e · 11aa48e
1 parent 5f4985b
commit 11aa48e
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}")