Remove redundant initialization of b_q_seq_len

WANDY666 · web-flow · commit d70d188b494b · 2025-08-29T10:22:13.000+08:00
diff --git a/lightllm/common/basemodel/triton_kernel/gen_decode_params.py b/lightllm/common/basemodel/triton_kernel/gen_decode_params.py
@@ -19,7 +19,6 @@ def gen_decode_params(b_seq_len: torch.Tensor):
             b_q_seq_len[: len(b_seq_len) // mtp_size], b_kv_seq_len[mtp_size - 1 :: mtp_size]
         )
     else:
-        b_q_seq_len = torch.ones_like(b_seq_len)
         b1_cu_q_seq_len, b1_cu_kv_seq_len = gen_cumsum_pad0_tensor(b_q_seq_len, b_kv_seq_len)
 
     return b_q_seq_len, b1_cu_q_seq_len, b_kv_seq_len, b1_cu_kv_seq_len, position_ids

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,6 @@ def gen_decode_params(b_seq_len: torch.Tensor):`
`19`	`19`	`b_q_seq_len[: len(b_seq_len) // mtp_size], b_kv_seq_len[mtp_size - 1 :: mtp_size]`
`20`	`20`	`)`
`21`	`21`	`else:`
`22`		`- b_q_seq_len = torch.ones_like(b_seq_len)`
`23`	`22`	`b1_cu_q_seq_len, b1_cu_kv_seq_len = gen_cumsum_pad0_tensor(b_q_seq_len, b_kv_seq_len)`
`24`	`23`
`25`	`24`	`return b_q_seq_len, b1_cu_q_seq_len, b_kv_seq_len, b1_cu_kv_seq_len, position_ids`