[Bug] fix import and unit test (#25558)

Signed-off-by: Jonas M. Kübler <44084297+jmkuebler@users.noreply.github.com>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Jonas M. Kübler 2025-09-24 12:17:59 +02:00 committed by yewentao256
parent d5c0f43b86
commit f98d4d38c0

View File

@ -11,7 +11,7 @@ from vllm.v1.attention.backends.utils import (UBatchSlice,
slice_query_start_locs,
split_attn_metadata,
split_decodes_and_prefills)
from vllm.v1.worker.ubatch_utils import create_ubatch_slices
from vllm.v1.worker.ubatch_splitting import create_ubatch_slices
@pytest.fixture