From 0452a58a32ac4a320b40fb6e6508c0e56e9f27e8 Mon Sep 17 00:00:00 2001
From: Balaram Buddharaju <169953907+brb-nv@users.noreply.github.com>
Date: Fri, 9 Jan 2026 13:25:22 -0800
Subject: [PATCH] append group_entries[0]

Signed-off-by: Balaram Buddharaju <169953907+brb-nv@users.noreply.github.com>
---
 tensorrt_llm/_torch/pyexecutor/executor_request_queue.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tensorrt_llm/_torch/pyexecutor/executor_request_queue.py b/tensorrt_llm/_torch/pyexecutor/executor_request_queue.py
index 50439d7b03..1930f51f35 100644
--- a/tensorrt_llm/_torch/pyexecutor/executor_request_queue.py
+++ b/tensorrt_llm/_torch/pyexecutor/executor_request_queue.py
@@ -384,8 +384,7 @@ class ExecutorRequestQueue:
                     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(