Revert D53075378: Disable exported_program.__call__

Differential Revision:
D53075378

Original commit changeset: e17671b655b6

Original Phabricator Diff: D53075378

fbshipit-source-id: abceb41683d6a831cbec152740477d2f10d1c582

Author: huydhn

exir/backend/test/test_backends_lifted.py

@@ -626,7 +626,7 @@ def forward(self, x_raw, h, c):
             ),
         )
 
-        new_res = program_with_delegates.exported_program().module()(*inputs)
+        new_res = program_with_delegates.exported_program()(*inputs)
         for t1, t2 in zip(new_res, orig_res, strict=True):
             self.assertTrue(torch.allclose(t1, t2, atol=1e-03, rtol=1e-03))
 
@@ -745,7 +745,7 @@ def forward(self, x_raw, h, c):
             HTAPartitionerOnePatternDemo()
         )
 
-        new_res = traced_with_delegate.exported_program().module()(*inputs)
+        new_res = traced_with_delegate.exported_program()(*inputs)
         for t1, t2 in zip(new_res, orig_res, strict=True):
             self.assertTrue(torch.allclose(t1, t2, atol=1e-03, rtol=1e-03))
 
@@ -768,7 +768,7 @@ def forward(self, x_raw, h, c):
         #     config=exir.ExecutorchBackendConfig(extract_delegate_segments=extract_delegate_segments),
         # )
 
-        new_res = program_with_delegates.exported_program().module()(*inputs)
+        new_res = program_with_delegates.exported_program()(*inputs)
         for t1, t2 in zip(new_res, orig_res, strict=True):
             self.assertTrue(torch.allclose(t1, t2, atol=1e-03, rtol=1e-03))
 
@@ -1029,7 +1029,7 @@ def f(x, y):
         partitioned = orig
         partitioned = partitioned.to_backend(AddMulPartitionerDemo())
 
-        new_res = partitioned.exported_program().module()(*inputs)
+        new_res = partitioned.exported_program()(*inputs)
         self.assertTrue(torch.allclose(orig_res, new_res[0]))
 
         toplevel_lowered = get_lowered_submodules(
@@ -1102,7 +1102,7 @@ def f(xs, y):
             map_fn_lowered[0][1].original_module.graph_module.code
         )
 
-        new_res = partitioned.exported_program().module()(*inputs)
+        new_res = partitioned.exported_program()(*inputs)
 
         self.assertTrue(torch.allclose(orig_res, new_res[0]))
 
@@ -1153,7 +1153,7 @@ def f(xs, pred1, pred2, y):
         partitioned = orig
         partitioned = partitioned.to_backend(AddMulPartitionerDemo())
 
-        new_res = partitioned.exported_program().module()(*inputs)
+        new_res = partitioned.exported_program()(*inputs)
         self.assertTrue(torch.allclose(orig_res, new_res[0]))
 
         toplevel_lowered = get_lowered_submodules(
@@ -1224,7 +1224,7 @@ def forward(self, x: List[torch.Tensor]):
                 return self.lowered(x)
 
         gm = to_edge(export(ComposedM(), inputs))
-        gm.exported_program().module()(*inputs)
+        gm.exported_program()(*inputs)
 
     def test_dict_input(self):
         def f(x: Dict[str, torch.Tensor]):
@@ -1246,4 +1246,4 @@ def forward(self, x: List[torch.Tensor]):
                 return self.lowered(x)
 
         gm = to_edge(export(ComposedM(), inputs))
-        gm.exported_program().module()(*inputs)
+        gm.exported_program()(*inputs)

exir/tests/test_verification.py

@@ -136,7 +136,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
         exec_prog = to_edge(export(model2, (inputs,))).to_executorch()
 
         exported_prog = exec_prog.exported_program()
-        res = exported_prog.module()(inputs)[0]  # noqa
+        res = exported_prog(inputs)[0]  # noqa
         # Verifiers are run internally in to_edge, export, and to_executorch.
         # If we make it this far then no errors were thrown in verification