from . import modeling_qwen3_vl_moe
def apply_qwen3vl_moe_npu_patch():
from mindspeed_mm.fsdp.ops.npu_patch import npu_fused_operator
modeling_qwen3_vl_moe.apply_rotary_pos_emb_vision = npu_fused_operator.apply_transformers_vision_rope_half_npu
modeling_qwen3_vl_moe.apply_rotary_pos_emb = npu_fused_operator.apply_transformers_rope_half_npu
modeling_qwen3_vl_moe.Qwen3VLMoeTextRMSNorm.forward = npu_fused_operator.rms_norm_forward_npu
modeling_qwen3_vl_moe.Qwen3VLMoeTextExperts.forward = npu_fused_operator.fused_moe_forward_npu