Skip to content

Commit 8973189

Browse files
clean code
Signed-off-by: shikang-hangzhou <459956190@qq.com>
1 parent 0453b2c commit 8973189

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm_ascend/multistream/ms_split.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,9 @@ def model_input_split_v1_mla_attn(
7272
attn_metadata.query_lens):
7373
return [attn_metadata]
7474

75-
query_start_loc_cpu: Any = np.zeros(shape=(len(attn_metadata.query_lens) + 1, ),
76-
dtype=int)
75+
query_start_loc_cpu: Any = np.zeros(shape=(len(attn_metadata.query_lens) +
76+
1, ),
77+
dtype=int)
7778
np.cumsum(attn_metadata.query_lens, out=query_start_loc_cpu[1:])
7879
if attn_metadata.num_prefills > 0:
7980
prefill_query_start_loc: Any = np.zeros(

0 commit comments

Comments
 (0)