You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
Copy file name to clipboardExpand all lines: projects/pt1/e2e_testing/main.py
+4-4Lines changed: 4 additions & 4 deletions
Original file line number
Diff line number
Diff line change
@@ -86,12 +86,12 @@ def _get_argparse():
86
86
"-c",
87
87
"--config",
88
88
choices=config_choices,
89
-
default="linalg",
89
+
default="fx_importer",
90
90
help=f"""
91
91
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.
95
95
"native_torch": run the torch.nn.Module as-is without compiling (useful for verifying model is deterministic; ALL tests should pass in this configuration).
96
96
"torchscript": compile the model to a torch.jit.ScriptModule, and then run that as-is (useful for verifying TorchScript is modeling the program correctly).
97
97
"lazy_tensor_core": run the model through the Lazy Tensor Core frontend and execute the traced graph.
0 commit comments