Skip to content

Commit 15818ac

Browse files
authored
ci: add test-backend-ops test for CPU (ggml-org#19268)
1 parent bf38346 commit 15818ac

2 files changed

Lines changed: 29 additions & 2 deletions

File tree

.github/workflows/build.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1532,7 +1532,7 @@ jobs:
15321532
- name: Test
15331533
id: ggml-ci
15341534
run: |
1535-
LLAMA_ARG_THREADS=$(nproc) bash ./ci/run.sh ./tmp/results ./tmp/mnt
1535+
LLAMA_ARG_THREADS=$(nproc) GG_BUILD_HIGH_PERF=1 bash ./ci/run.sh ./tmp/results ./tmp/mnt
15361536
15371537
ggml-ci-arm64-cpu-high-perf:
15381538
runs-on: ubuntu-22.04-arm
@@ -1558,7 +1558,7 @@ jobs:
15581558
- name: Test
15591559
id: ggml-ci
15601560
run: |
1561-
LLAMA_ARG_THREADS=$(nproc) GG_BUILD_NO_SVE=1 GG_BUILD_NO_BF16=1 GG_BUILD_EXTRA_TESTS_0=1 bash ./ci/run.sh ./tmp/results ./tmp/mnt
1561+
LLAMA_ARG_THREADS=$(nproc) GG_BUILD_HIGH_PERF=1 GG_BUILD_NO_SVE=1 GG_BUILD_NO_BF16=1 GG_BUILD_EXTRA_TESTS_0=1 bash ./ci/run.sh ./tmp/results ./tmp/mnt
15621562
15631563
ggml-ci-arm64-cpu-high-perf-sve:
15641564
runs-on: ubuntu-22.04-arm

ci/run.sh

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -635,6 +635,29 @@ function gg_check_build_requirements {
635635
fi
636636
}
637637

638+
function gg_run_test_backend_ops_cpu {
639+
cd ${SRC}
640+
641+
cd build-ci-release
642+
643+
set -e
644+
645+
(time ./bin/test-backend-ops -b CPU ) 2>&1 | tee -a $OUT/${ci}-test-backend-ops-cpu.log
646+
647+
set +e
648+
}
649+
650+
function gg_sum_test_backend_ops_cpu {
651+
gg_printf '### %s\n\n' "${ci}"
652+
653+
gg_printf 'Runs test-backend-ops for CPU backend\n'
654+
gg_printf '- status: %s\n' "$(cat $OUT/${ci}.exit)"
655+
gg_printf '```\n'
656+
gg_printf '%s\n' "$(cat $OUT/${ci}-test-backend-ops-cpu.log)"
657+
gg_printf '```\n'
658+
gg_printf '\n'
659+
}
660+
638661
## main
639662

640663
export LLAMA_LOG_PREFIX=1
@@ -663,6 +686,10 @@ ret=0
663686
test $ret -eq 0 && gg_run ctest_debug
664687
test $ret -eq 0 && gg_run ctest_release
665688

689+
if [ ! -z ${GG_BUILD_HIGH_PERF} ]; then
690+
test $ret -eq 0 && gg_run test_backend_ops_cpu
691+
fi
692+
666693
if [ -z ${GG_BUILD_LOW_PERF} ]; then
667694
test $ret -eq 0 && gg_run embd_bge_small
668695
test $ret -eq 0 && gg_run rerank_tiny

0 commit comments

Comments
 (0)