Explorar o código

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

ggml-ci
Georgi Gerganov %!s(int64=2) %!d(string=hai) anos
pai
achega
1a8c8795d6
Modificáronse 1 ficheiros con 6 adicións e 4 borrados
  1. 6 4
      ci/run.sh

+ 6 - 4
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