We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
reduce.py
1 parent 39eec89 commit 3f4ac9fCopy full SHA for 3f4ac9f
python/triton_kernels/reduce.py
python/triton_kernels/triton_kernels/reduce.py
@@ -147,6 +147,8 @@ def reduce(
147
Returns:
148
- output: torch.Tensor
149
The reduced tensor with `dim` removed.
150
+ - output_mxscale: Optional[torch.Tensor]
151
+ The output mx scale if input is micro-scaled, else None.
152
"""
153
if x.ndim != 3:
154
raise NotImplementedError("reduce only supports 3D inputs in this implementation")
0 commit comments