Fix the rebase to upstream/main wrong import and format

Signed-off-by: leo-pony <[email protected]>

Author: leo-pony

vllm_ascend/models/qwen3_next.py

@@ -16,7 +16,7 @@
 from vllm.distributed import (divide, get_tensor_model_parallel_rank,
                               get_tensor_model_parallel_world_size)
 from vllm.forward_context import get_forward_context
-from vllm.model_executor.layers.fla.ops.chunk import chunk_gated_delta_rule, RMSNormGated, chunk
+from vllm.model_executor.layers.fla.ops import chunk, chunk_gated_delta_rule
 from vllm.model_executor.layers.fla.ops.fused_recurrent import \
     fused_recurrent_gated_delta_rule
 from vllm.model_executor.layers.fused_moe import FusedMoE
@@ -579,7 +579,7 @@ def _forward_core(
             mixed_qkv_spec = mixed_qkv_spec.view(
                 attn_metadata.num_spec_decodes, -1, mixed_qkv_spec.size(-1))
             mixed_qkv_spec = rearrange(mixed_qkv_spec, 'b l d -> b d l')
-            mixed_qkv_spec = causal_conv1d_update(
+            mixed_qkv_spec = causal_conv1d.causal_conv1d_update(
                 mixed_qkv_spec,
                 conv_state,
                 conv_weights,
@@ -596,7 +596,7 @@ def _forward_core(
         if attn_metadata.num_prefills > 0:
             # - "cache_indices" updates the conv_state cache in positions
             #   pointed to by "mamba_cache_params.state_indices_tensor"
-            mixed_qkv_non_spec = causal_conv1d_fn(
+            mixed_qkv_non_spec = causal_conv1d.causal_conv1d_fn(
                 mixed_qkv_non_spec.transpose(0, 1),
                 conv_weights,
                 self.conv1d.bias,
@@ -607,7 +607,7 @@ def _forward_core(
                 query_start_loc=non_spec_query_start_loc,
             ).transpose(0, 1)
         elif attn_metadata.num_decodes > 0:
-            mixed_qkv_non_spec = causal_conv1d_update(
+            mixed_qkv_non_spec = causal_conv1d.causal_conv1d_update(
                 mixed_qkv_non_spec,
                 conv_state,
                 conv_weights,