From 27be0bddc23d7df60eebe2c21ee655692e06d41c Mon Sep 17 00:00:00 2001 From: Sage Moore Date: Fri, 6 Dec 2024 20:15:57 +0000 Subject: [PATCH] format --- vllm/compilation/activation_quant_fusion.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vllm/compilation/activation_quant_fusion.py b/vllm/compilation/activation_quant_fusion.py index 146f77546a7c3..c073d0fb7f767 100644 --- 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)