From 6f5def360073b74940d01fc4236713dd174fdd1f Mon Sep 17 00:00:00 2001 From: Bin Bao Date: Tue, 23 Jan 2024 06:07:05 -0800 Subject: [PATCH] Fix a bug in the torch._export.aot_load API (#2128) Summary: X-link: https://github.com/pytorch/torchrec/pull/1652 X-link: https://github.com/pytorch/pytorch/pull/118039 tree_flatten_spec should use args instead of *args clone of https://github.com/pytorch/pytorch/pull/117948 but with some fbcode specific changes Reviewed By: angelayi Differential Revision: D52982401 --- userbenchmark/dynamo/dynamobench/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py index 26a6624087..b8512af247 100644 --- a/userbenchmark/dynamo/dynamobench/common.py +++ b/userbenchmark/dynamo/dynamobench/common.py @@ -1149,7 +1149,7 @@ def export_aot_inductor(model, example_inputs, device): def opt_aot_inductor(_, example_inputs, collect_outputs=False): example_args, example_kwargs = _normalize_bench_inputs(example_inputs) - return optimized(example_args, example_kwargs) + return optimized(*example_args, **example_kwargs) return opt_aot_inductor