Skip to content

feat: show progress bar during multi-turn ISL pre-compute#329

Closed
YibaiMeng wants to merge 2 commits into
mlcommons:mainfrom
YibaiMeng:fix/multiturn-isl-progress
Closed

feat: show progress bar during multi-turn ISL pre-compute#329
YibaiMeng wants to merge 2 commits into
mlcommons:mainfrom
YibaiMeng:fix/multiturn-isl-progress

Conversation

@YibaiMeng

@YibaiMeng YibaiMeng commented Jun 2, 2026

Copy link
Copy Markdown

What does this PR do?

Show a tqdm progress bar over the multi-turn ISL pre-computation loop in _precompute_isl_for_multi_turn (src/inference_endpoint/commands/benchmark/execute.py).

Why: On large multi-turn datasets the serial apply_chat_template loop runs for minutes with no output (30 min on a 26k-turn agentic-coding dataset, with the entire benchmark taking less than 50 minutes), so it looks like a hang. This PR adds a progress bar.

Type of change

  • Bug fix
  • New feature
  • Documentation update
  • Refactor/cleanup

Related issues

N/A

Testing

  • Tests added/updated
  • All tests pass locally
  • Manual testing completed

Checklist

  • Code follows project style
  • Pre-commit hooks pass
  • Documentation updated (if needed)

The serial apply_chat_template loop over every client turn runs silent
for minutes on large multi-turn datasets, so it looks like a hang. Show a
throttled tqdm bar (2s mininterval) over the loop for liveness + ETA in
captured logs. No behavior change.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
@github-actions

github-actions Bot commented Jun 2, 2026

Copy link
Copy Markdown

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a tqdm progress bar to track the progress of pre-computing ISL token counts during multi-turn benchmarks. The feedback recommends disabling the progress bar when INFO logging is not enabled to avoid cluttering logs in non-interactive or CI/CD environments.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Comment thread src/inference_endpoint/commands/benchmark/execute.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
@YibaiMeng YibaiMeng marked this pull request as ready for review June 2, 2026 22:08
@YibaiMeng YibaiMeng requested a review from a team June 2, 2026 22:08
@tianmu-li

Copy link
Copy Markdown
Collaborator

Hi @YibaiMeng, could you check if #336 alleviates the issue?

@YibaiMeng

Copy link
Copy Markdown
Author

Closing this PR as it has been addressed in #349 and #336

@YibaiMeng YibaiMeng closed this Jun 9, 2026
@github-actions github-actions Bot locked and limited conversation to collaborators Jun 9, 2026
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants