[test] refine fx_importer_backend with verbose option (#4084)

* make `verbose` (aka `-v`) option's behavior to be same with other
backends. `verbose=True` would print the IR of each stage.
* clear `verbose` and `enable_ir_printing` option.

Author: qingyunqu

projects/pt1/python/torch_mlir_e2e_test/configs/fx_importer_backend.py

@@ -42,6 +42,7 @@ def __init__(self, backend, output_type="linalg-on-tensors", torch_compile=False
     def compile(
         self, program: torch.nn.Module, verbose: bool = False
     ) -> torch.nn.Module:
+        self._verbose = verbose
         return program
 
     def run(self, artifact: torch.nn.Module, trace: Trace):
@@ -84,6 +85,7 @@ def _base_backend(gm: torch.fx.GraphModule, example_inputs):
                     gm,
                     output_type=self._output_type,
                     model_name=artifact.__class__.__name__,
+                    verbose=self._verbose,
                 )
                 module = self._backend.compile(module)
                 backend_module = self._backend.load(module)
@@ -128,6 +130,7 @@ def _export_run(self, artifact: torch.nn.Module, trace: Trace) -> Trace:
                 # While the current e2e tests don't exercise symbolic shapes,
                 # enabling this here ensures they don't regress either.
                 import_symbolic_shape_expressions=True,
+                verbose=self._verbose,
             )
             module = self._backend.compile(module)
             backend_module = self._backend.load(module)

python/torch_mlir/fx.py

@@ -26,15 +26,18 @@
 
 def _module_lowering(
     verbose,
+    enable_ir_printing,
     output_type,
     torch_mod,
     extra_library_file_name=None,
     backend_legal_ops=None,
 ):
+    if verbose:
+        print("\n====================")
+        print("TorchFX IR")
+        print(torch_mod)
 
     if output_type == OutputType.RAW:
-        if verbose:
-            print(torch_mod)
         return torch_mod
     # TODO: pass extra_library_file_name by caller
 
@@ -59,7 +62,7 @@ def _module_lowering(
         torch_mod,
         f"builtin.module(func.func(torch-match-quantized-custom-ops), torchdynamo-export-to-torch-backend-pipeline{option_string})",
         "Lowering TorchFX IR -> Torch Backend IR",
-        enable_ir_printing=verbose,
+        enable_ir_printing=enable_ir_printing,
     )
     return lower_mlir_module(verbose, output_type, torch_mod)
 
@@ -76,6 +79,7 @@ def export_and_import(
     decomposition_table: Optional[Dict[torch._ops.OperatorBase, Callable]] = None,
     func_name: str = "main",
     enable_graph_printing: bool = False,
+    verbose: bool = False,
     enable_ir_printing: bool = False,
     backend_legal_ops: Optional[list[str]] = None,
     **kwargs,
@@ -115,6 +119,7 @@ def export_and_import(
         )
 
     return _module_lowering(
+        verbose,
         enable_ir_printing,
         OutputType.get(output_type),
         fx_importer.module,
@@ -129,6 +134,7 @@ def stateless_fx_import(
     hooks: Optional[FxImporterHooks] = None,
     model_name: str = "main",
     enable_graph_printing: bool = False,
+    verbose: bool = False,
     enable_ir_printing: bool = False,
     backend_legal_ops: Optional[list[str]] = None,
 ):
@@ -140,6 +146,7 @@ def stateless_fx_import(
         fx_importer = FxImporter(context=context, hooks=hooks)
     fx_importer.import_stateless_graph(gm.graph, func_name=model_name)
     return _module_lowering(
+        verbose,
         enable_ir_printing,
         OutputType.get(output_type),
         fx_importer.module,