Skip to content

feat: add --ep support to e2e eval scripts#395

Merged
DingmaomaoBJTU merged 1 commit into
mainfrom
qiowu/fix_e2e_script
Apr 27, 2026
Merged

feat: add --ep support to e2e eval scripts#395
DingmaomaoBJTU merged 1 commit into
mainfrom
qiowu/fix_e2e_script

Conversation

@DingmaomaoBJTU

Copy link
Copy Markdown
Collaborator

Summary

Thread --ep through the e2e evaluation pipeline so it can target any execution provider.

  • run_eval.py: new --ep arg forwarded to winml config/perf/eval
  • run_sa_eval.py: new --ep/--device args forwarded to all SA stages
  • sa_comparison.py: parameterize EP in run_sa_with_info and parse_sa_json (no longer hardcodes QNNExecutionProvider)

Usage

# E2E perf with DML on GPU
python scripts/e2e_eval/run_eval.py --device gpu --ep dml --hf-model microsoft/resnet-50

# SA eval with OpenVINO
python scripts/e2e_eval/run_sa_eval.py --ep OpenVINOExecutionProvider --device NPU

Depends on #392 for the analyzer-side EP support.

🤖 Generated with Claude Code

Thread --ep through run_eval.py, run_sa_eval.py, and sa_comparison.py
so the evaluation pipeline can target any EP (e.g. --ep dml --device gpu).

- run_eval.py: new --ep arg forwarded to winml config/perf/eval
- run_sa_eval.py: new --ep/--device args forwarded to all stages
- sa_comparison.py: parameterize EP in run_sa_with_info and parse_sa_json
@DingmaomaoBJTU DingmaomaoBJTU requested a review from a team as a code owner April 24, 2026 10:26
@timenick

Copy link
Copy Markdown
Collaborator

Do we currently have basic unit tests for the EP and device combination within the analyze component?

@DingmaomaoBJTU

Copy link
Copy Markdown
Collaborator Author

Do we currently have basic unit tests for the EP and device combination within the analyze component?

#392 More uts here

@DingmaomaoBJTU DingmaomaoBJTU merged commit 8cca1fd into main Apr 27, 2026
16 checks passed
@DingmaomaoBJTU DingmaomaoBJTU deleted the qiowu/fix_e2e_script branch April 27, 2026 02:58
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.

3 participants