Skip to content

Commit

Permalink
Fix torch_onnx patch accuracy bug in benchmark (#130586)
Browse files Browse the repository at this point in the history
Summary:
The ONNX related compilers have another route of accuracy check, and this PR brings torch_onnx compiler to the right measurement.

X-link: pytorch/pytorch#130586
Approved by: https://github.com/justinchuby

Reviewed By: izaitsevfb

Differential Revision: D59724163

fbshipit-source-id: 5bdaa15e433d1908a20011d1ea6c370da9242850
  • Loading branch information
titaiwangms authored and facebook-github-bot committed Jul 14, 2024
1 parent 35489f1 commit 8359e1a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion userbenchmark/dynamo/dynamobench/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 8359e1a

Please sign in to comment.