tiny fix

Signed-off-by: wxsIcey <[email protected]>

Author: wxsIcey

tests/e2e/singlecard/compile/backend.py

@@ -16,12 +16,15 @@
 #
 from copy import deepcopy
 from typing import Any, Callable, List, Optional, Sequence
+
 import torch.fx as fx
 from torch._inductor.decomposition import select_decomp_table
 from vllm.compilation.fx_utils import OpOverload
 from vllm.config import get_current_vllm_config
+
 from vllm_ascend.compilation.compiler_interface import compile_fx
 
+
 class TestBackend:
     """
     A custom compilation backend for testing operator fusion passes.
@@ -121,4 +124,4 @@ def check_after_ops(self, ops: Sequence[OpOverload]):
         for op in ops:
             num_post = len(self.find_nodes_by_target(self.graph_post_pass, op))
             print(f"Op {op}: post={num_post}")
-            assert num_post > 0, f"Op {op} not found in post-pass graph"
+            assert num_post > 0, f"Op {op} not found in post-pass graph"

tests/e2e/singlecard/compile/test_norm_quant_fusion.py

@@ -21,10 +21,12 @@
 import torch.nn as nn
 import torch_npu
 import vllm.config
-from vllm.config import VllmConfig, ModelConfig
 from vllm.compilation.fx_utils import OpOverload
+from vllm.config import ModelConfig, VllmConfig
+
 from tests.e2e.singlecard.compile.backend import TestBackend
-from vllm_ascend.compilation.passes.norm_quant_fusion_pass import AddRMSNormQuantFusionPass
+from vllm_ascend.compilation.passes.norm_quant_fusion_pass import \
+    AddRMSNormQuantFusionPass
 
 
 class TestModel(nn.Module):
@@ -89,7 +91,8 @@ def test_rmsnorm_quant_fusion(dtype: torch.dtype, hidden_size: int,
     vllm_config = VllmConfig(model_config=ModelConfig(dtype=dtype))
 
     with vllm.config.set_current_vllm_config(vllm_config):
-        backend = TestBackend(custom_passes=[AddRMSNormQuantFusionPass(vllm_config=vllm_config)])
+        backend = TestBackend(
+            custom_passes=[AddRMSNormQuantFusionPass(vllm_config=vllm_config)])
         model = TestModel(hidden_size, eps, device="npu")
         model = model.to("npu")
 

vllm_ascend/compilation/graph_fusion_pass_manager.py

@@ -47,6 +47,7 @@ def configure(self, config: VllmConfig):
         self.ascend_compilation_config: dict = config.additional_config.get(
             "ascend_compilation_config", {})
         if self.ascend_compilation_config.get("fuse_norm_quant", True):
-            from .passes.norm_quant_fusion_pass import AddRMSNormQuantFusionPass
+            from .passes.norm_quant_fusion_pass import \
+                AddRMSNormQuantFusionPass
             self.passes.append(AddRMSNormQuantFusionPass(config))
         # Add more passes here as needed