From 0da4f9bb49d6e1d4ef01fae5dcb2cb8dccaa7ca6 Mon Sep 17 00:00:00 2001
From: Jhen-Jie Hong <developer@jhen.me>
Date: Sun, 19 Jan 2025 13:08:49 +0800
Subject: [PATCH] feat: add CMAKE_CUDA_ARCHITECTURES

---
 scripts/build-linux.sh | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/scripts/build-linux.sh b/scripts/build-linux.sh
index 10ed296..f520af5 100755
--- 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