[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 <noreply@anthropic.com>
Harry Mellor committed
Sep 25, 2026 at 12:01 UTC
afea5c20c7c20d697132f6145cb68e06d447c23a
1 file changed
+1
-1
vllm/model_executor/layers/fusion/mm_input_norm.py
+1
-1
@@ -71,7 +71,7 @@ def fused_mm_input_norm_triton(
71
outputs: torch.Tensor,
72
weight: torch.Tensor,
73
bias: torch.Tensor,
74
-):
74
+) -> torch.Tensor:
75
"""Fused per-channel affine transform for normalisation.
76
77
Args: