Skip to content

Commit

Permalink
Remove example inputs from aoti_compile_and_package (#2544)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #2544

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

Test Plan: Imported from OSS

Reviewed By: yushangdi

Differential Revision: D66136724

Pulled By: angelayi

fbshipit-source-id: d779e017e387e40922b362d33c7af310f764eae7
  • Loading branch information
angelayi authored and facebook-github-bot committed Nov 20, 2024
1 parent 75c0f31 commit c7420bb
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 c7420bb

Please sign in to comment.