OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ONNX] Fix torch_onnx patch accuracy bug in benchmark (#130586)

Browse Source 
The ONNX related compilers have another route of accuracy check, and this PR brings torch_onnx compiler to the right measurement.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/130586
Approved by: https://github.com/justinchuby
This commit is contained in:
titaiwangms 2024-07-12 15:47:57 +00:00 committed by PyTorch MergeBot
parent e5657024b5
commit 18418a7dbb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/dynamo/common.py
View File

@ -4005,7 +4005,7 @@ def run(runner, args, original_dir=None):
)
experiment = speedup_experiment_onnx
output_filename = "torch_onnx_patch.csv"
current_onnx_compiler = "torch_onnx_patch"
current_onnx_compiler = "dynamo"
elif args.dynamo_onnx:
optimize_ctx = functools.partial(
optimize_onnx_ctx,