[BugFix]: Fix sglang.bench_one_batch (#16925)

Co-authored-by: Xinyuan Tong <115166877+JustinTong0323@users.noreply.github.com>
2026-01-16 13:15:34 -08:00
parent a7f5677abe
commit 7f8353aff3
1 changed files with 1 additions and 1 deletions
--- a/python/sglang/bench_one_batch.py
+++ b/python/sglang/bench_one_batch.py
@@ -316,7 +316,7 @@ def prepare_extend_inputs_for_correctness_test(
        req.fill_ids += input_ids[i][bench_args.cut_len :]
        req.prefix_indices = model_runner.req_to_token_pool.req_to_token[
            i, : 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))
    return reqs