Skip to content

Commit 42c3c29

Browse files
authored
[e2e] : Change default to fx_importer. (#4387)
Since torchscript path is not maintained anymore, updating the default choice for e2e testing to `fx_importer` from `linalg`. Backend being tested is still the `linalg` path. Fixes #4386
1 parent ba513ee commit 42c3c29

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

docs/development.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -507,8 +507,8 @@ The following assumes you are in the `projects/pt1` directory:
507507
./tools/e2e_test.sh
508508
# Run tests that match the regex `Conv2d`, with verbose errors.
509509
./tools/e2e_test.sh --filter Conv2d --verbose
510-
# Run tests on the TOSA backend.
511-
./tools/e2e_test.sh --config tosa
510+
# Run tests on the TOSA backend via fx_importer path
511+
./tools/e2e_test.sh --config fx_importer_tosa
512512
```
513513
514514
Alternatively, you can run the tests via Python directly:

projects/pt1/e2e_testing/main.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -86,12 +86,12 @@ def _get_argparse():
8686
"-c",
8787
"--config",
8888
choices=config_choices,
89-
default="linalg",
89+
default="fx_importer",
9090
help=f"""
9191
Meaning of options:
92-
"linalg": run through torch-mlir"s default Linalg-on-Tensors backend.
93-
"tosa": run through torch-mlir"s default TOSA backend.
94-
"stablehlo": run through torch-mlir"s default Stablehlo backend.
92+
"linalg": run the model through JIT importer frontend with torchscript (deprecated) and execute the graph using Linalg-on-Tensors backend.
93+
"tosa": run the model through JIT importer frontend with torchscript (deprecated) and execute the graph using TOSA backend.
94+
"stablehlo": run the model through JIT importer frontend with torchscript (deprecated) and execute the graph using Stablehlo backend.
9595
"native_torch": run the torch.nn.Module as-is without compiling (useful for verifying model is deterministic; ALL tests should pass in this configuration).
9696
"torchscript": compile the model to a torch.jit.ScriptModule, and then run that as-is (useful for verifying TorchScript is modeling the program correctly).
9797
"lazy_tensor_core": run the model through the Lazy Tensor Core frontend and execute the traced graph.

0 commit comments

Comments
 (0)