Function nvfp4_mmq_smem 

pub const fn nvfp4_mmq_smem(mmq_x: u32) -> u32

Dynamic shared memory for a given M-tile, from the vendor’s layout: ids_dst[mmq_x] + y-tile[mmq_x * MMQ_TILE_Y_K(=36) ints, padded to 256] + x-tile[128 * MMQ_MMA_TILE_X_K_FP4(=76) ints], all 4-byte. Reproduces the previously-hardcoded 57856 at mmq_x=128, which is the check that this derivation matches the kernel’s actual layout.