Skip to content

Commit af85566

Browse files
author
weijinqian_v1
committed
handle code clean
Signed-off-by: weijinqian_v1 <weijinqian@huawei.com>
1 parent 5956ef0 commit af85566

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

vllm_ascend/models/qwen3_dbo.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -267,11 +267,9 @@ def _forward_ms_layer_alltoallv_finegrained(
267267
] * num_micro_batchs
268268
tokens_per_expert = [None] * num_micro_batchs
269269
dispatched_input = [None] * num_micro_batchs
270-
shared_expert_output = [None] * num_micro_batchs
271270
router_expert_output = [None] * num_micro_batchs
272271
chunked_hidden_states_sizes = [None] * num_micro_batchs
273272
token_dispatchers = self.mlp.experts.token_dispatchers
274-
has_shared_expert = hasattr(self.mlp, 'shared_experts')
275273

276274
def discard_tensor(tensor):
277275
if isinstance(tensor, torch.Tensor):

vllm_ascend/multistream/ms_split.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ def model_input_split_v1_attn(
293293
token_index)
294294

295295
is_only_prefill_pre = is_only_prefill_post = attn_metadata.is_only_prefill
296-
has_prefill_pre, has_prefill_post = torch.any(
296+
has_prefill_pre, _ = torch.any(
297297
query_lens_pre > 1).item(), torch.any(query_lens_post > 1).item()
298298

299299
if not attn_metadata.is_only_prefill:

0 commit comments

Comments
 (0)