flashinfer.comm.compute_fp4_swizzled_layout_sf_size

flashinfer.comm.compute_fp4_swizzled_layout_sf_size(total_row: int, total_column: int) → int

Compute the padded size (rows times columns) of the FP4 swizzled layout.

Parameters:

• total_row (int) – Logical row count of the un-padded layout.

• total_column (int) – Logical column count of the un-padded layout.

Returns:

padded_row * padded_column for the swizzled layout.

Return type:

int