From 57a8aaa07846efe40e9d214bf9c1508602ab288b Mon Sep 17 00:00:00 2001 From: Travis Johnson Date: Wed, 26 Nov 2025 14:36:30 -0700 Subject: [PATCH] fix test_compare_graphs_chunked_prefill Signed-off-by: Travis Johnson --- tests/aftu/test_compare_graphs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/aftu/test_compare_graphs.py b/tests/aftu/test_compare_graphs.py index e810befc..7d6f4250 100644 --- a/tests/aftu/test_compare_graphs.py +++ b/tests/aftu/test_compare_graphs.py @@ -231,11 +231,11 @@ def test_compare_graphs_chunked_prefill(model: ModelInfo, max_num_seqs: int, monkeypatch.setenv("TORCH_SENDNN_CACHE_ENABLE", "0") monkeypatch.setenv('VLLM_DT_CHUNK_LEN', str(chunk_size)) - monkeypatch.setenv('VLLM_SPYRE_USE_CHUNKED_PREFILL', "1") patch_environment(use_cb=True, warmup_shapes=None, backend="sendnn", - monkeypatch=monkeypatch) + monkeypatch=monkeypatch, + use_chunked_prefill=True) original_cwd = os.getcwd() try: