From 8fa80b072c7a888b418170d6904e11c6f7bb7173 Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Fri, 5 Dec 2025 12:16:43 +0900
Subject: [PATCH] chore: improve determine patams

---
 build_linux.sh | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/build_linux.sh b/build_linux.sh
index 0819c3b..d64551b 100755
--- a/build_linux.sh
+++ b/build_linux.sh
@@ -56,15 +56,22 @@ if [[ -z "${MAX_JOBS:-}" && -z "${NVCC_THREADS:-}" ]]; then
   # Calculate max product based on following constraints:
   # - MAX_JOBS x NVCC_THREADS(<= 4) <= NUM_THREADS
   # - 2.5GB x MAX_JOBS x NVCC_THREADS(<= 4) <= RAM_GB
+
+  # Set MAX_PRODUCT from RAM
   MAX_PRODUCT_CPU=$NUM_THREADS
-  MAX_PRODUCT_RAM=$(awk -v ram="$RAM_GB" 'BEGIN {print int(ram / 2.25)}')
+  MAX_PRODUCT_RAM=$(awk -v ram="$RAM_GB" 'BEGIN {print int(ram / 2.5)}')
   MAX_PRODUCT=$((MAX_PRODUCT_CPU < MAX_PRODUCT_RAM ? MAX_PRODUCT_CPU : MAX_PRODUCT_RAM))
 
   # Set MAX_JOBS and NVCC_THREADS so that MAX_JOBS x NVCC_THREADS ≈ MAX_PRODUCT with NVCC_THREADS <= 4
   BASE_THREADS=$(awk -v max="$MAX_PRODUCT" 'BEGIN {print int(sqrt(max))}')
-  if (( BASE_THREADS <= 4 )); then
-    MAX_JOBS=$BASE_THREADS
+
+  if awk "BEGIN {exit !($RAM_GB <= 16)}"; then
+    # If RAM is less than 16GB, set NVCC_THREADS to 1
+    NVCC_THREADS=1
+    MAX_JOBS=$((MAX_PRODUCT / NVCC_THREADS))
+  elif (( BASE_THREADS <= 4 )); then
     NVCC_THREADS=$BASE_THREADS
+    MAX_JOBS=$BASE_THREADS
   else
     NVCC_THREADS=4
     MAX_JOBS=$((MAX_PRODUCT / NVCC_THREADS))