From 8652001b6a176feaf9478aaf65b31ee51a9289a2 Mon Sep 17 00:00:00 2001
From: chenxb002 <22113491+chenxb002@users.noreply.github.com>
Date: Fri, 29 May 2026 17:34:42 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20use=20req.req=5Fpool=5Fidx=20instead=20o?=
 =?UTF-8?q?f=20loop=20variable=20for=20req=5Fto=5Ftoken=20i=E2=80=A6=20(#2?=
 =?UTF-8?q?6534)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: kjuuii <1375341936@qq.com>
---
 python/sglang/bench_one_batch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/sglang/bench_one_batch.py b/python/sglang/bench_one_batch.py
index 69ec7c32e..a8b2e59e1 100644
--- a/python/sglang/bench_one_batch.py
+++ b/python/sglang/bench_one_batch.py
@@ -386,7 +386,7 @@ def prepare_extend_inputs_for_correctness_test(
         req.fill_ids += input_ids[i][bench_args.cut_len :]
         if model_runner is not None:
             req.prefix_indices = model_runner.req_to_token_pool.req_to_token[
-                i, : bench_args.cut_len
+                req.req_pool_idx, : bench_args.cut_len
             ].to(req.prefix_indices.dtype)
             req.logprob_start_len = -1
             req.set_extend_input_len(len(req.fill_ids) - len(req.prefix_indices))