File tree Expand file tree Collapse file tree 1 file changed +0
-13
lines changed
torchao/prototype/mx_formats Expand file tree Collapse file tree 1 file changed +0
-13
lines changed Original file line number Diff line number Diff line change @@ -1394,19 +1394,6 @@ def triton_scale_swizzle(
1394
1394
BLOCK_ROWS : tl .constexpr ,
1395
1395
BLOCK_COLS : tl .constexpr ,
1396
1396
):
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
- """
1410
1397
pid_row = tl .program_id (0 )
1411
1398
pid_col = tl .program_id (1 )
1412
1399
You can’t perform that action at this time.
0 commit comments