From afea5c20c7c20d697132f6145cb68e06d447c23a Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Fri, 25 Sep 2026 12:01:11 +0200 Subject: [PATCH] [Docs] Add return annotation to `fused_mm_input_norm_triton` (#58687) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: Claude Opus 5.5 --- vllm/model_executor/layers/fusion/mm_input_norm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/fusion/mm_input_norm.py b/vllm/model_executor/layers/fusion/mm_input_norm.py index ea1dd1da22..a21828033d 100644 --- a/vllm/model_executor/layers/fusion/mm_input_norm.py +++ b/vllm/model_executor/layers/fusion/mm_input_norm.py @@ -71,7 +71,7 @@ def fused_mm_input_norm_triton( outputs: torch.Tensor, weight: torch.Tensor, bias: torch.Tensor, -): +) -> torch.Tensor: """Fused per-channel affine transform for normalisation. Args: