[PD] improve kv offset calculation for MHA model with different tp size#18163

ShangmingCai merged 3 commits intosgl-project:mainsgl-project/sglang:mainfrom

Ch3ngY1:feat/numpy-optimizeCh3ngY1/sglang:feat/numpy-optimizeCopy head branch name to clipboard