remove args

Author: angelayi

examples/models/llama/export_llama_lib.py

@@ -800,6 +800,11 @@ def _export_llama(args) -> LLMEdgeManager:  # noqa: C901
             # pyre-fixme[16]: Module `backends` has no attribute `qualcomm`.
             canonicalize_program(builder.edge_manager.exported_program())
 
+        print("lfq: exported program after to_backend, graph_module")
+        print(builder.edge_manager.exported_program().graph_module)
+        print("lfq: exported program after to_backend, print_readable")
+        print(builder.edge_manager.exported_program().graph_module.print_readable())
+
         builder = builder.to_executorch()
 
     if args.profile_memory:

examples/models/llama3_2_vision/text_decoder/test/test_text_decoder.py

@@ -74,8 +74,6 @@ def test_llama3_2_text_decoder_aoti(self) -> None:
         with tempfile.TemporaryDirectory() as tmpdir:
             path = torch._inductor.aoti_compile_and_package(
                 ep,
-                model.get_example_inputs(),
-                kwargs=model.get_example_kwarg_inputs(),
                 package_path=os.path.join(tmpdir, "text_decoder.pt2"),
             )
             encoder_aoti = torch._inductor.aoti_load_package(path)

examples/models/llama3_2_vision/vision_encoder/test/test_vision_encoder.py

@@ -36,7 +36,6 @@ def test_flamingo_vision_encoder(self) -> None:
         with tempfile.TemporaryDirectory() as tmpdir:
             path = torch._inductor.aoti_compile_and_package(
                 ep,
-                model.get_example_inputs(),
                 package_path=os.path.join(tmpdir, "vision_encoder.pt2"),
             )
             print(path)

extension/llm/export/builder.py

@@ -193,6 +193,7 @@ def export(self) -> "LLMEdgeManager":
                     dynamic_shapes=dynamic_shape,
                     strict=True,
                 )
+                exported_module = exported_module.run_decompositions({})
             else:
                 logging.info("Exporting with:")
                 logging.info(f"inputs: {self.example_inputs}")

extension/llm/modules/test/test_position_embeddings.py

@@ -177,7 +177,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)