(sf, rows, cols, scaling_vector_size=16)
| 226 | |
| 227 | @torch.library.register_fake("trtllm::reswizzle_sf") |
| 228 | def _(sf, rows, cols, scaling_vector_size=16): |
| 229 | sf_cols = ceil_div(cols, scaling_vector_size) |
| 230 | padded_rows, padded_sf_cols = compute_swizzled_sf_shape(rows, sf_cols) |
| 231 | num_partitions = sf.numel() // (padded_rows * padded_sf_cols) |
| 232 | total_rows = num_partitions * rows |
| 233 | sz = pad_up(total_rows, 128) * pad_up(cols, 4) |
| 234 | return sf.new_empty(sz) |
| 235 | |
| 236 | |
| 237 | def next_positive_power_of_2(x: int) -> int: |
nothing calls this directly
no test coverage detected