Remove example inputs from aoti_compile_and_package

angelayi · web-flow · commit c1522489cbcb · 2025-01-10T14:43:19.000-08:00
Differential Revision: D67998952 Pull Request resolved: #2244
diff --git a/tests/torchtune/modules/_export/test_export_position_embeddings.py b/tests/torchtune/modules/_export/test_export_position_embeddings.py
@@ -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)

Original file line number	Diff line number	Diff line change
`@@ -161,7 +161,6 @@ def test_tiled_token_positional_embedding_aoti(self):`
`161`	`161`	`with tempfile.TemporaryDirectory() as tmpdir:`
`162`	`162`	`path = torch._inductor.aoti_compile_and_package(`
`163`	`163`	`tpe_ep,`
`164`		`- (self.x, self.aspect_ratio),`
`165`	`164`	`package_path=os.path.join(tmpdir, "tpe.pt2"),`
`166`	`165`	`)`
`167`	`166`	`tpe_aoti = load_package(path)`