Skip to content

fix: enhance model evaluation preparation for non-task-based evaluation - #2468

Merged
wenhuach21 merged 2 commits into
mainfrom
xinhe/10-10a
Oct 10, 2026
Merged

wenhuach21 merged 2 commits into
mainfrom
xinhe/10-10a

Conversation

@xin3he

@xin3he xin3he commented Oct 10, 2026

Copy link
Copy Markdown
Contributor

Description

model is not dispatched to device, fix it.

Type of Change

Bug fix

Related Issues

Fixes or relates to #

Checklist Before Submitting

  • My code has been tested locally.
  • Documentation has been updated as needed.
  • New or updated tests are included where applicable.
  • The CUDA CI has passed. You can trigger it by commenting /azp run Unit-Test-CUDA-AutoRound.

Copilot AI balanced review requested due to automatic review settings October 10, 2026 03:29

Copilot AI 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.

🟡 Changes recommended

The new dispatch branch lacks regression coverage.

1 open finding
What changed in this PR

Ensures in-memory models are dispatched to the configured device before standard evaluation.

Changes:

  • Prepares model dtype and device placement for non-task-based evaluation.
  • Preserves task-by-task evaluation’s existing dispatch behavior.
File Description
auto_round/​eval/​evaluation.py Dispatches passed-in models before evaluation.

🧠 Review effort: Balanced


💡 Configure MCP servers for context-aware, tailored reviews. Learn more in the docs.

Comment thread auto_round/eval/evaluation.py
Signed-off-by: Xin He <xin3.he@intel.com>
@chensuyue chensuyue added this to the 0.16.1 milestone Oct 10, 2026
@AutoRoundBot

Copy link
Copy Markdown
Collaborator

/azp run Unit-Test-CUDA-AutoRound

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 1 pipeline(s).

@wenhuach21
wenhuach21 merged commit 1f421d6 into main Oct 10, 2026
51 checks passed
@wenhuach21
wenhuach21 deleted the xinhe/10-10a branch October 10, 2026 06:07
XuehaoSun pushed a commit that referenced this pull request Oct 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants