There was an error while loading. Please reload this page.
1 parent 7414d25 commit 4489b6aCopy full SHA for 4489b6a
tests/v1/worker/test_gpu_model_runner.py
@@ -951,6 +951,7 @@ def test_hybrid_block_table_initialization():
951
max_num_reqs = 10
952
max_num_blocks_per_req = 20
953
max_num_batched_tokens = 512
954
+ dcp_kv_cache_interleave_size = 8
955
956
block_table = BlockTable(
957
block_size=block_size,
@@ -960,6 +961,7 @@ def test_hybrid_block_table_initialization():
960
961
pin_memory=False,
962
device=torch.device(DEVICE),
963
kernel_block_size=kernel_block_sizes[0],
964
+ dcp_kv_cache_interleave_size=dcp_kv_cache_interleave_size,
965
)
966
967
# Verify hybrid block configuration
0 commit comments