Fix torch_onnx patch accuracy bug in benchmark (#130586)

titaiwangms · facebook-github-bot · commit 8359e1aa553c · 2024-07-14T12:59:54.000-07:00
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
diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py
@@ -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,

Original file line number	Diff line number	Diff line change
`@@ -4005,7 +4005,7 @@ def run(runner, args, original_dir=None):`
`4005`	`4005`	`)`
`4006`	`4006`	`experiment = speedup_experiment_onnx`
`4007`	`4007`	`output_filename = "torch_onnx_patch.csv"`
`4008`		`- current_onnx_compiler = "torch_onnx_patch"`
	`4008`	`+ current_onnx_compiler = "dynamo"`
`4009`	`4009`	`elif args.dynamo_onnx:`
`4010`	`4010`	`optimize_ctx = functools.partial(`
`4011`	`4011`	`optimize_onnx_ctx,`