bf16_grouped_tc_shape_ok

metrale_model_layers::layers::ops

Function bf16_grouped_tc_shape_ok 

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

2026-10-02: Whether the BF16 tensor-core kernels take an n-column, k-deep projection: k in whole load groups (two 32-K chunks; 128 for margin), n in whole CTAs of geometry.