rafactor: move `gqa_group_size` from template parameter to input arguments

[yzh119]

262 is out of sync with main, this PR rebased the code on main branch.

This PR also greatly reduce the binary size because we don't need to compile prefill kernels for each gqa group size.