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 5
5
set -ex
6
6
7
7
# 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; }
9
9
trap remove_docker_container EXIT
10
10
remove_docker_container
11
11
12
12
# 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
14
38
You can’t perform that action at this time.
0 commit comments