Skip to content

Commit

Permalink
Remove example inputs from aoti_compile_and_package
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/pytorch#140991




cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang aakhundov

imported-using-ghimport

Reviewed By: yushangdi

Differential Revision: D66136724

Pulled By: angelayi
  • Loading branch information
angelayi authored and facebook-github-bot committed Nov 19, 2024
1 parent 75c0f31 commit d7bc2a9
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions userbenchmark/dynamo/dynamobench/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit d7bc2a9

Please sign in to comment.