fp8_grouped_tc_shape_ok

metrale_model_layers::layers::ops

Function fp8_grouped_tc_shape_ok 

Source 
pub fn fp8_grouped_tc_shape_ok(
    n: u32,
    k: u32,
    geometry: Fp8GroupedGeometry,
) -> bool
Expand description

2026-09-28: Whether the tensor-core kernels take an n-column, k-deep projection: both in whole 128 blocks (the FP8 scale blocks) and n in whole CTAs of geometry.