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

ggml-ci
This commit is contained in:
Georgi Gerganov 2023-10-12 13:44:56 +03:00 committed by GitHub
parent b016596d90
commit 1a8c8795d6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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