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: The args were removed in pytorch/pytorch#140991

Differential Revision: D67998952
  • Loading branch information
angelayi authored and facebook-github-bot committed Jan 10, 2025
1 parent e420bc0 commit 395d941
Showing 1 changed file with 0 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,6 @@ def test_tiled_token_positional_embedding_aoti(self):
with tempfile.TemporaryDirectory() as tmpdir:
path = torch._inductor.aoti_compile_and_package(
tpe_ep,
(self.x, self.aspect_ratio),
package_path=os.path.join(tmpdir, "tpe.pt2"),
)
tpe_aoti = load_package(path)
Expand Down

0 comments on commit 395d941

Please sign in to comment.