From 60fc0688b08cfa3689d62a520d7fd93b33f4a46b Mon Sep 17 00:00:00 2001 From: Sayan Saha Date: Sun, 23 Nov 2025 20:28:55 -0500 Subject: [PATCH 1/3] [e2e] : Change default to fx_importer. --- projects/pt1/e2e_testing/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/pt1/e2e_testing/main.py b/projects/pt1/e2e_testing/main.py index ae4986b7f96c..bd8d9ec46a8c 100644 --- a/projects/pt1/e2e_testing/main.py +++ b/projects/pt1/e2e_testing/main.py @@ -86,7 +86,7 @@ def _get_argparse(): "-c", "--config", choices=config_choices, - default="linalg", + default="fx_importer", help=f""" Meaning of options: "linalg": run through torch-mlir"s default Linalg-on-Tensors backend. From 3aecdaa789fdbff78ab00582f1c8af5766d8f0c1 Mon Sep 17 00:00:00 2001 From: Sayan Saha Date: Mon, 24 Nov 2025 08:24:00 -0500 Subject: [PATCH 2/3] [e2e] : Update doc to point to fx_importer_tosa. --- docs/development.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/development.md b/docs/development.md index 1c85437376b1..f1584d25e1ba 100644 --- a/docs/development.md +++ b/docs/development.md @@ -507,8 +507,8 @@ The following assumes you are in the `projects/pt1` directory: ./tools/e2e_test.sh # Run tests that match the regex `Conv2d`, with verbose errors. ./tools/e2e_test.sh --filter Conv2d --verbose -# Run tests on the TOSA backend. -./tools/e2e_test.sh --config tosa +# Run tests on the TOSA backend via fx_importer path +./tools/e2e_test.sh --config fx_importer_tosa ``` Alternatively, you can run the tests via Python directly: From 1ffc4602e8895517dacdb32bda81a18fe317084c Mon Sep 17 00:00:00 2001 From: Sayan Saha Date: Mon, 24 Nov 2025 14:08:18 -0500 Subject: [PATCH 3/3] [Task] : Update help text for e2e.main --- projects/pt1/e2e_testing/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/projects/pt1/e2e_testing/main.py b/projects/pt1/e2e_testing/main.py index bd8d9ec46a8c..e95faabeb91d 100644 --- a/projects/pt1/e2e_testing/main.py +++ b/projects/pt1/e2e_testing/main.py @@ -89,9 +89,9 @@ def _get_argparse(): default="fx_importer", help=f""" Meaning of options: -"linalg": run through torch-mlir"s default Linalg-on-Tensors backend. -"tosa": run through torch-mlir"s default TOSA backend. -"stablehlo": run through torch-mlir"s default Stablehlo backend. +"linalg": run the model through JIT importer frontend with torchscript (deprecated) and execute the graph using Linalg-on-Tensors backend. +"tosa": run the model through JIT importer frontend with torchscript (deprecated) and execute the graph using TOSA backend. +"stablehlo": run the model through JIT importer frontend with torchscript (deprecated) and execute the graph using Stablehlo backend. "native_torch": run the torch.nn.Module as-is without compiling (useful for verifying model is deterministic; ALL tests should pass in this configuration). "torchscript": compile the model to a torch.jit.ScriptModule, and then run that as-is (useful for verifying TorchScript is modeling the program correctly). "lazy_tensor_core": run the model through the Lazy Tensor Core frontend and execute the traced graph.