File tree Expand file tree Collapse file tree 1 file changed +26
-2
lines changed
.buildkite/scripts/hardware_ci Expand file tree Collapse file tree 1 file changed +26
-2
lines changed Original file line number Diff line number Diff line change 55set -ex
66
77# Setup cleanup
8- remove_docker_container () { docker rm -f cpu-test || true ; docker system prune -f; }
8+ remove_docker_container () { podman rm -f cpu-test-ubi9-ppc || true ; podman system prune -f; }
99trap remove_docker_container EXIT
1010remove_docker_container
1111
1212# Try building the docker image
13- docker build -t cpu-test -f docker/Dockerfile.ppc64le .
13+ podman build -t cpu-test-ubi9-ppc -f docker/Dockerfile.ppc64le .
14+
15+ # Run the image
16+ podman run -itd --entrypoint /bin/bash -v /tmp/:/root/.cache/huggingface --privileged=true --network host -e HF_TOKEN --name cpu-test-ubi9-ppc cpu-test-ubi9-ppc
17+
18+ function cpu_tests() {
19+
20+ # offline inference
21+ podman exec cpu-test-ubi9-ppc bash -c "
22+ set -e
23+ python3 examples/offline_inference/basic/generate.py --model facebook/opt-125m"
24+
25+ # Run basic model test
26+ podman exec cpu-test-ubi9-ppc bash -c "
27+ set -e
28+ pip install pytest pytest-asyncio einops peft Pillow soundfile transformers_stream_generator matplotlib
29+ pip install sentence-transformers datamodel_code_generator
30+ pytest -v -s tests/models/embedding/language/test_cls_models.py::test_classification_models[float-jason9693/Qwen2.5-1.5B-apeach]
31+ pytest -v -s tests/models/embedding/language/test_embedding.py::test_models[half-BAAI/bge-base-en-v1.5]
32+ pytest -v -s tests/models/encoder_decoder/language -m cpu_model"
33+ }
34+
35+ # All of CPU tests are expected to be finished less than 40 mins.
36+ export -f cpu_tests
37+ timeout 40m bash -c cpu_tests
1438
You can’t perform that action at this time.
0 commit comments