pub fn shard_fp8_block_scaled(
src: &Fp8Weight,
kind: TpShardKind,
tp_rank: usize,
tp_size: usize,
block_size: usize,
gpu: &dyn GpuBackend,
) -> Result<Fp8Weight>Expand description
2026-09-25: Shard an FP8 block-scaled weight. weight is [N, K] FP8 bytes;
row_scale is [N/block_size, K/block_size] FP32. Both slice on the same
axis at block granularity. Same ownership rule as shard_quantized_nvfp4.