Skip to content

Commit 0692a61

Browse files
committed
fix lint
Co-authored-by: realliujiaxu <realliujiaxu@163.com> Signed-off-by: zhaozx-cn <zhaozx2116@163.com>
1 parent 86110a0 commit 0692a61

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm_ascend/ops/common_fused_moe.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,9 @@ def forward_impl(self, hidden_states: torch.Tensor,
320320
forward_context = get_forward_context()
321321
moe_comm_method_name = forward_context.moe_comm_method_name
322322

323-
if hasattr(self, '_shared_experts') and forward_context.flashcomm_v1_enabled:
323+
if not hasattr(
324+
self,
325+
'_shared_experts') and forward_context.flashcomm_v1_enabled:
324326
hidden_states = torch.ops.vllm.maybe_all_gather_and_maybe_unpad(
325327
hidden_states, True)
326328
router_logits = torch.ops.vllm.maybe_all_gather_and_maybe_unpad(

0 commit comments

Comments
 (0)