Skip to content

Commit 4149c58

Browse files
add uts for multistream
1 parent 7603645 commit 4149c58

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/ut/multistream/test_ms_split.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
from vllm_ascend.multistream.base import MSAttentionMetadataSplitConfig
77
from vllm_ascend.multistream.ms_split import (compute_split_seq_index,
88
model_input_split_v1_mla_attn,
9-
split_attn_tensor_type,
10-
split_attn_int_type)
9+
split_attn_int_type,
10+
split_attn_tensor_type)
1111

1212

1313
class TestMsSplit(unittest.TestCase):

0 commit comments

Comments
 (0)