From 5973675bc30d7fc85d131b5c61ac4a2318b16fcf Mon Sep 17 00:00:00 2001 From: Ke Bao Date: Wed, 30 Jul 2025 02:03:02 +0800 Subject: [PATCH] Fix moe align kernel test (#8531) --- sgl-kernel/tests/test_moe_align.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sgl-kernel/tests/test_moe_align.py b/sgl-kernel/tests/test_moe_align.py index 8d35e75c1..550c7a1ad 100644 --- a/sgl-kernel/tests/test_moe_align.py +++ b/sgl-kernel/tests/test_moe_align.py @@ -229,7 +229,7 @@ def test_moe_align_block_size_compare_implementations( matching_indices = torch.where(expert_ids_cuda == expert_idx)[0] block_sorted_start = matching_indices[0].item() * block_size block_sorted_end = min( - (matching_indices[-1].item() + 1) * block_size, max_num_tokens_padded + (matching_indices[-1].item() + 1) * block_size, num_tokens_post_pad_cuda.item() ) selected_sorted_ids_cuda = sorted_ids_cuda[