[KVCache][Test] Fix TIR attn kernels for uncommon group size#17074

yongwww merged 1 commit intoapache:mainapache/tvm:mainfrom

MasterJH5574:tvm-dev/2024-06-08-attn-kernel-group-sizeMasterJH5574/tvm:tvm-dev/2024-06-08-attn-kernel-group-sizeCopy head branch name to clipboard