feat: support more head dim in RoPE kernel#2109

raayandhar wants to merge 3 commits intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

raayandhar:users/rdhar/rope_head_dimraayandhar/flashinfer:users/rdhar/rope_head_dimCopy head branch name to clipboard