add tileN = 8,16 for SM120 blockscale GEMM.

[b8zhong]

It will be for use with SwapAB.

[b8zhong]

Hi @depaulmillz , I was wondering if you could take a look at this PR? Since I noticed you were the last one to add TileN = 32. Thanks!

[depaulmillz]

Awesome. Have you been able to try with group GEMM as well?

[b8zhong]

@depaulmillz Yes. Technically, it works (this PR is also compatible with group GEMM changes as well). But for example when testing on two common cases, DSR1 TP = 8 and Qwen-3 MoE TP = 1, the speedup can only be 3-5% for BS = 1. So it's faster (as expected), but not by much.

[depaulmillz]

It looks like you will need to add an assertion to prevent compiling ping-pong with MMA_N=8 which will expect a (2,2,1) layout shape for the MMA. I saw some ref check errors when testing the MR on pingpong MMA_N=8 kernels due to this.

[b8zhong]

@depaulmillz Just added a guard for it. Could you take a try now?

[b8zhong]

Hi @depaulmillz , just bumping this PR for visibility. Thanks a lot!

[depaulmillz]

@hwu36 can we merge this?

[b8zhong]

Thanks @depaulmillz @hwu36 !