From 7ddbb0e2d4178c02df5069bc87eaf3e197ce8097 Mon Sep 17 00:00:00 2001 From: zhuofan1123 Date: Tue, 13 Jan 2026 07:31:44 +0000 Subject: [PATCH] fix: prevent block_ids overlap in generate_request_pair --- tests/test_utils.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index 7099f66356..9907ac8b01 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -68,10 +68,7 @@ def test_config(request: pytest.FixtureRequest): def generate_request_pair(idx: int, block_per_request, num_gpu_blocks, tokens_per_block, dp_size): """Generate a request pair with token_ids, block_ids, and dp_id""" start_idx = (idx * block_per_request) % num_gpu_blocks - if start_idx + block_per_request >= num_gpu_blocks: - start_idx = ( - (start_idx + block_per_request) % num_gpu_blocks - ) + assert start_idx + block_per_request <= num_gpu_blocks block_ids = torch.arange( start_idx, start_idx + block_per_request,