From: Georgi Gerganov <redacted>
Date: Thu, 12 Oct 2023 10:44:56 +0000 (+0300)
Subject: ci : check if there is enough VRAM (#3596)
X-Git-Tag: upstream/0.0.4488~3114
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=1a8c8795d64b04df96c28f29faac2d6e256f53bc;p=pkg%2Fggml%2Fsources%2Fllama.cpp

ci : check if there is enough VRAM (#3596)

ggml-ci
---

diff --git a/ci/run.sh b/ci/run.sh
index 942b2e00..34c9129c 100755
--- a/ci/run.sh
+++ b/ci/run.sh
@@ -496,10 +496,12 @@ test $ret -eq 0 && gg_run ctest_debug
 test $ret -eq 0 && gg_run ctest_release
 
 if [ -z ${GG_BUILD_LOW_PERF} ]; then
-    if [ -z ${GG_BUILD_CUDA} ]; then
-        test $ret -eq 0 && gg_run open_llama_3b_v2
-    else
-        test $ret -eq 0 && gg_run open_llama_7b_v2
+    if [ -z ${GG_BUILD_VRAM_GB} ] || [ ${GG_BUILD_VRAM_GB} -ge 8 ]; then
+        if [ -z ${GG_BUILD_CUDA} ]; then
+            test $ret -eq 0 && gg_run open_llama_3b_v2
+        else
+            test $ret -eq 0 && gg_run open_llama_7b_v2
+        fi
     fi
 fi