Skip to content

Commit 5128a75

Browse files
authored
Fix stale _paged_kv_cache reference in test_paged_prefix_caching (#207)
1 parent b5a1595 commit 5128a75

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_paged_prefix_caching.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def _make_paged_runner(num_layers: int = 2) -> mr.MetalModelRunner:
2626
runner = mr.MetalModelRunner.__new__(mr.MetalModelRunner)
2727
runner.model = MagicMock()
2828
runner._is_stt = False
29-
runner._paged_kv_cache = MagicMock() # non-None enables paged path
29+
runner._paged_attention_backend = MagicMock() # non-None enables paged path
3030
runner._paged_block_size = 4
3131
runner._paged_request_seq_lens = {}
3232
runner._request_states = {}

0 commit comments

Comments
 (0)