append group_entries[0]

Author: brb-nv

tensorrt_llm/_torch/pyexecutor/executor_request_queue.py

@@ -384,8 +384,7 @@ def _fetch_new_requests_attention_dp(
                     f"CP ranks within DP group {dp_group_idx} have mismatched request counts: " \
                     f"{[entry[0] for entry in group_entries]}"
                 # Use token count from cp_rank0.
-                aggregated_responses.append(
-                    [group_entries[0][0], group_entries[0][1]])
+                aggregated_responses.append(group_entries[0])
             responses_list = aggregated_responses
         else:
             responses_list = self.dist.tp_allgather(