[aoti] Remove example inputs from aoti_compile_and_package (pytorch#144520)

angelayi · pytorchmergebot · commit a94ec0a9a5ac · 2025-01-10T21:56:23.000Z
Summary: The args were removed in pytorch#140991 Test Plan: CI Differential Revision: D67998954 Pull Request resolved: pytorch#144520 Approved by: https://github.com/yushangdi
diff --git a/test/export/test_draft_export.py b/test/export/test_draft_export.py
@@ -417,7 +417,6 @@ def forward(self, x, y):
         with tempfile.NamedTemporaryFile(suffix=".pt2") as f:
             torch._inductor.aoti_compile_and_package(
                 draft_ep,
-                example_inputs,
                 package_path=f.name,
             )
 
diff --git a/test/inductor/test_minifier.py b/test/inductor/test_minifier.py
@@ -239,9 +239,7 @@ def forward(self, inp, *, k):
     ep = torch.export.export(
         model, example_inputs, kwargs
     )
-    torch._inductor.aoti_compile_and_package(
-        ep, example_inputs, kwargs
-    )
+    torch._inductor.aoti_compile_and_package(ep)
 """
         return self._run_full_test(run_code, None, expected_error, isolate=True)
 

Original file line number	Diff line number	Diff line change
`@@ -417,7 +417,6 @@ def forward(self, x, y):`
`417`	`417`	`with tempfile.NamedTemporaryFile(suffix=".pt2") as f:`
`418`	`418`	`torch._inductor.aoti_compile_and_package(`
`419`	`419`	`draft_ep,`
`420`		`- example_inputs,`
`421`	`420`	`package_path=f.name,`
`422`	`421`	`)`
`423`	`422`