diff options
| author | Georgi Gerganov <ggerganov@gmail.com> | 2023-10-12 13:44:56 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-12 13:44:56 +0300 |
| commit | 1a8c8795d64b04df96c28f29faac2d6e256f53bc (patch) | |
| tree | a93802c509d9332f937250ece7c09e0960a0a3d3 | |
| parent | b016596d903641f8825cd94bb6742e1de0c21017 (diff) | |
ci : check if there is enough VRAM (#3596)
ggml-ci
| -rwxr-xr-x | ci/run.sh | 10 |
1 files changed, 6 insertions, 4 deletions
@@ -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 |