From 2e5d0821eed0859d444d6c7ac16c2b69206edf06 Mon Sep 17 00:00:00 2001 From: Bin Bao Date: Tue, 23 Jan 2024 04:09:00 -0800 Subject: [PATCH] Fix a bug in the torch._export.aot_load API (#2128) Summary: 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