Refactor the topk parallelization part for the routing kernels#5567

byshiue merged 1 commit intoNVIDIA:mainNVIDIA/TensorRT-LLM:mainfrom

ChristinaZ:refactor_topK_in_routing_trtllmgenChristinaZ/TensorRT-LLM:refactor_topK_in_routing_trtllmgenCopy head branch name to clipboard