diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py index 508672ba4..9727b9191 100644 --- a/userbenchmark/dynamo/dynamobench/common.py +++ b/userbenchmark/dynamo/dynamobench/common.py @@ -1521,9 +1521,7 @@ def load(cls, model, example_inputs): strict=False, ) with torch.no_grad(): - package_path = torch._inductor.aoti_compile_and_package( - ep, example_args, example_kwargs - ) # type: ignore[arg-type] + package_path = torch._inductor.aoti_compile_and_package(ep) # type: ignore[arg-type] cls.cache[key] = torch._inductor.aoti_load_package(package_path)