feat: add CMAKE_CUDA_ARCHITECTURES

mybigday · Jan 19, 2025 · 0da4f9b · 0da4f9b
1 parent 3c12e9a
commit 0da4f9b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/scripts/build-linux.sh b/scripts/build-linux.sh
@@ -12,7 +12,10 @@ if [ $ARCH == "x86_64" ]; then
 
   # Check CUDA is available
   if [ -f /usr/local/cuda/bin/nvcc ]; then
-    yarn clean && yarn build-native --CDLLAMA_CUDA=1 --CDVARIANT=cuda
+    yarn clean && yarn build-native \
+      --CDLLAMA_CUDA=1 \
+      --CDVARIANT=cuda \
+      --CDCMAKE_CUDA_ARCHITECTURES=89 # > GeForce RTX 40 series
   else
     echo "CUDA is not available, skipping CUDA build"
   fi
@@ -22,7 +25,10 @@ else
 
   # Check CUDA is available
   if [ -f /usr/local/cuda/bin/nvcc ]; then
-    yarn clean && yarn build-native --CDLLAMA_CUDA=1 --CDVARIANT=cuda
+    yarn clean && yarn build-native \
+      --CDLLAMA_CUDA=1 \
+      --CDVARIANT=cuda \
+      --CDCMAKE_CUDA_ARCHITECTURES=87 # > Jetson Orin series
   else
     echo "CUDA is not available, skipping CUDA build"
   fi