[QST] Where is CuTe ValLayout in TiledMMA as of CUTLASS 3.4.0

[ZhZhang711]

I find that make_tiled_mma no longer takes MMAValLayout as its argument, but the cute document is still using a MMAValLayout to construct a TiledMMA. I wonder where does the MMAValLayout go and how can I replicate a TiledMMA across values. Thanks!

[ccecka]

From the documentation:

The above represents a 16x16x4 MMA now, but we can immediately expand this "tile size" up to 32x32x4 instead:

TiledMMA mma = make_tiled_mma(SM70_8x8x4_F32F16F16F32_NT{},
                              Layout<Shape <_2,_2>,
                                     Stride<_2,_1>>{},  // 2x2 n-major layout of Atoms
                              Tile<_32,_32,_4>{});      // 32x32x4 tiler
print_latex(mma);

The Tiler parameter expands the TiledMMA in values. The documentation is up to date and does not mention the prior MMAValLayout.

[ZhZhang711]

Oh I see, thanks. I thought that Tile<_32,_32,_4>{} was an MMAValLayout, but in fact it is a Permutations.
So if I need TiledShape_MNK that existed prior to CUTLASS 3.4.0, am I expected to call tile_size_mnk thrice for I = 1,2,3?