improve to find cuda for linux arm64 #5317

xmake-io · Jul 9, 2024 · 85f26eb · 85f26eb
1 parent 46952b0
commit 85f26eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xmake/modules/detect/sdks/find_cuda.lua b/xmake/modules/detect/sdks/find_cuda.lua
@@ -105,7 +105,7 @@ function _find_cuda(sdkdir)
     if is_host("windows") then
         local subdir = is_arch("x64") and "x64" or "Win32"
         table.insert(linkdirs, path.join(sdkdir, "lib", subdir))
-    elseif is_host("linux") and is_arch("x86_64") then
+    elseif is_host("linux") and is_arch("x86_64", "arm64") then
         table.insert(linkdirs, path.join(sdkdir, "lib64"))
     else
         table.insert(linkdirs, path.join(sdkdir, "lib"))