[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>
This commit is contained in:
co-authored by
Claude Opus 5.5
parent
378504a544
commit
afea5c20c7
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user