Skip to content

Commit 3d27302

Browse files
committed
ci : check if there is enough VRAM
ggml-ci
1 parent b016596 commit 3d27302

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

ci/run.sh

+6-4
Original file line numberDiff line numberDiff line change
@@ -496,10 +496,12 @@ test $ret -eq 0 && gg_run ctest_debug
496496
test $ret -eq 0 && gg_run ctest_release
497497

498498
if [ -z ${GG_BUILD_LOW_PERF} ]; then
499-
if [ -z ${GG_BUILD_CUDA} ]; then
500-
test $ret -eq 0 && gg_run open_llama_3b_v2
501-
else
502-
test $ret -eq 0 && gg_run open_llama_7b_v2
499+
if [ -z ${GG_BUILD_VRAM_GB} ] || [ ${GG_BUILD_VRAM_GB} -ge 8 ]; then
500+
if [ -z ${GG_BUILD_CUDA} ]; then
501+
test $ret -eq 0 && gg_run open_llama_3b_v2
502+
else
503+
test $ret -eq 0 && gg_run open_llama_7b_v2
504+
fi
503505
fi
504506
fi
505507

0 commit comments

Comments
 (0)