From 9ea7fc442a80027e8ade1d6b6efbb7f197ab18ab Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Fri, 5 Dec 2025 12:13:37 +0900
Subject: [PATCH] chore: add environment test

---
 .github/workflows/test-utils.yml | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/test-utils.yml b/.github/workflows/test-utils.yml
index e7bff5d..187dc9c 100644
--- a/.github/workflows/test-utils.yml
+++ b/.github/workflows/test-utils.yml
@@ -41,15 +41,22 @@ jobs:
             # 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 / 3.0)}')
+            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))