nvfp4_grouped_tc_shape_ok

metrale_model_layers::layers::ops

Function nvfp4_grouped_tc_shape_ok 

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

2026-10-02: Whether the tensor-core kernels take an n-column, k-deep projection: k in whole load groups of two 128-wide chunks, n in whole CTAs of geometry.