Skip to content

Commit 3f4ac9f

Browse files
authored
[KERNELS][NFC] Remove the redundant reduce.py file (#8524)
It's redundant with https://github.com/triton-lang/triton/blob/main/python/triton_kernels/triton_kernels/reduce.py
1 parent 39eec89 commit 3f4ac9f

File tree

2 files changed

+2
-282
lines changed

2 files changed

+2
-282
lines changed

python/triton_kernels/reduce.py

Lines changed: 0 additions & 282 deletions
This file was deleted.

python/triton_kernels/triton_kernels/reduce.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,8 @@ def reduce(
147147
Returns:
148148
- output: torch.Tensor
149149
The reduced tensor with `dim` removed.
150+
- output_mxscale: Optional[torch.Tensor]
151+
The output mx scale if input is micro-scaled, else None.
150152
"""
151153
if x.ndim != 3:
152154
raise NotImplementedError("reduce only supports 3D inputs in this implementation")

0 commit comments

Comments
 (0)