format

vllm-project · Dec 6, 2024 · 27be0bd · 27be0bd
1 parent 4f123d5
commit 27be0bd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vllm/compilation/activation_quant_fusion.py b/vllm/compilation/activation_quant_fusion.py
@@ -13,8 +13,9 @@
 logger = init_logger(__name__)
 
 
-def silu_mul_pattern_static(result: torch.Tensor, result_silu_mul: torch.Tensor,
-                            input: torch.Tensor, scale: torch.Tensor):
+def silu_mul_pattern_static(result: torch.Tensor,
+                            result_silu_mul: torch.Tensor, input: torch.Tensor,
+                            scale: torch.Tensor):
     at1 = auto_functionalized(torch.ops._C.silu_and_mul.default,
                               result=result_silu_mul,
                               input=input)