Skip to content

Commit a2c5ca1

Browse files
authored
Removing DocBlock to unblock MXFP4 w/ Unwrap Tensor (#2292)
stack-info: PR: #2292, branch: drisspg/stack/63
1 parent f0f1f6c commit a2c5ca1

File tree

1 file changed

+0
-13
lines changed

1 file changed

+0
-13
lines changed

torchao/prototype/mx_formats/kernels.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1394,19 +1394,6 @@ def triton_scale_swizzle(
13941394
BLOCK_ROWS: tl.constexpr,
13951395
BLOCK_COLS: tl.constexpr,
13961396
):
1397-
"""
1398-
Rearranges tensor data from row-major to block-scaled swizzle format.
1399-
1400-
Args:
1401-
scale_ptr: Pointer to the input scale tensor
1402-
scale_rows: Number of rows in the scale tensor
1403-
scale_cols: Number of columns in the scale tensor
1404-
output_ptr: Pointer to the output tensor
1405-
input_row_stride: Stride between rows in the input tensor
1406-
output_block_stride: Stride between blocks in the output tensor
1407-
BLOCK_ROWS: Number of rows in a tile (compile-time constant)
1408-
BLOCK_COLS: Number of columns in a tile (compile-time constant)
1409-
"""
14101397
pid_row = tl.program_id(0)
14111398
pid_col = tl.program_id(1)
14121399

0 commit comments

Comments
 (0)