Skip to content

Commit afcec8b

Browse files
sducouedicjoerunde
andauthored
[CI/CD] mark current tests that are failing when compile cache is enabled (#171)
* mark current tests that are failing with compile cache Signed-off-by: Sophie du Couédic <[email protected]> * ✨ update mark to automatically skip when caching enabled Signed-off-by: Joe Runde <[email protected]> --------- Signed-off-by: Sophie du Couédic <[email protected]> Signed-off-by: Joe Runde <[email protected]> Co-authored-by: Joe Runde <[email protected]>
1 parent c8c24ce commit afcec8b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/e2e/test_spyre_warmup_shapes.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,19 @@
33
Run `python -m pytest tests/test_spyre_warmup_shapes.py`.
44
"""
55

6+
import os
7+
68
import pytest
79
from spyre_util import (compare_results, generate_hf_output,
810
generate_spyre_vllm_output, get_spyre_backend_list,
911
get_spyre_model_list)
1012
from vllm import SamplingParams
1113

1214

15+
# temporary for filtering until bug with caching gets fixed
16+
@pytest.mark.skipif(
17+
os.environ.get("TORCH_SENDNN_CACHE_ENABLE") == "1",
18+
reason="torch_sendnn caching is currently broken with this configuration")
1319
@pytest.mark.parametrize("model", get_spyre_model_list())
1420
@pytest.mark.parametrize("prompts", [
1521
7 * [

0 commit comments

Comments
 (0)