Skip to content
This repository was archived by the owner on Oct 11, 2024. It is now read-only.

Commit 49e1f3c

Browse files
committed
fix softfail on cpu tests
1 parent ec8ef04 commit 49e1f3c

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

.buildkite/run-cpu-test-ppc64le.sh

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,13 @@ docker run -itd --entrypoint /bin/bash -v ~/.cache/huggingface:/root/.cache/hugg
1818
# Run basic model test
1919
docker exec cpu-test bash -c "
2020
pip install pytest matplotlib einops transformers_stream_generator
21-
pytest -v -s tests/models -m \"not vlm\" --ignore=tests/models/test_embedding.py --ignore=tests/models/test_oot_registration.py --ignore=tests/models/test_registry.py --ignore=tests/models/test_jamba.py --ignore=tests/models/test_danube3_4b.py" # Mamba and Danube3-4B on CPU is not supported
21+
pytest -v -s tests/models -m \"not vlm\" \
22+
--ignore=tests/models/test_embedding.py \
23+
--ignore=tests/models/test_oot_registration.py \
24+
--ignore=tests/models/test_registry.py \
25+
--ignore=tests/models/test_jamba.py \
26+
--ignore=tests/models/test_mamba.py \
27+
--ignore=tests/models/test_danube3_4b.py" # Mamba kernels and Danube3-4B on CPU is not supported
2228

2329
# online inference
2430
docker exec cpu-test bash -c "

.buildkite/run-cpu-test.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ docker exec cpu-test bash -c "
2727
pytest -v -s tests/models/decoder_only/language \
2828
--ignore=tests/models/test_fp8.py \
2929
--ignore=tests/models/decoder_only/language/test_jamba.py \
30+
--ignore=tests/models/decoder_only/language/test_mamba.py \
3031
--ignore=tests/models/decoder_only/language/test_granitemoe.py \
3132
--ignore=tests/models/decoder_only/language/test_danube3_4b.py" # Mamba and Danube3-4B on CPU is not supported
3233

0 commit comments

Comments
 (0)