From bc62a7fbf304fe548d6c6b7f8872a460f6941f5b Mon Sep 17 00:00:00 2001 From: Sean Silva Date: Thu, 30 Sep 2021 19:45:43 +0000 Subject: [PATCH] Update to new name of torchscript-module-to-linalg-on-tensors-backend-pipeline --- examples/resnet_inference.ipynb | 2 +- examples/torchfx/examples/example_add_tanh_sigmoid.py | 2 +- examples/torchscript_resnet18_e2e.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/resnet_inference.ipynb b/examples/resnet_inference.ipynb index dfca3c5b9..1038e1ad6 100644 --- a/examples/resnet_inference.ipynb +++ b/examples/resnet_inference.ipynb @@ -152,7 +152,7 @@ " mb.import_module(scripted._c, class_annotator)\n", "\n", " ## Lower the MLIR from TorchScript to RefBackend, passing through linalg-on-tensors.\n", - " pm = PassManager.parse('torchscript-to-linalg-on-tensors-backend-pipeline', mb.module.context)\n", + " pm = PassManager.parse('torchscript-module-to-linalg-on-tensors-backend-pipeline', mb.module.context)\n", " pm.run(mb.module)\n", "\n", " ## Invoke RefBackend to compile to compiled artifact form.\n", diff --git a/examples/torchfx/examples/example_add_tanh_sigmoid.py b/examples/torchfx/examples/example_add_tanh_sigmoid.py index 07cbc13d4..d9a78144e 100644 --- a/examples/torchfx/examples/example_add_tanh_sigmoid.py +++ b/examples/torchfx/examples/example_add_tanh_sigmoid.py @@ -41,7 +41,7 @@ torch_mlir_module.dump() print(torch_mlir_module.operation.verify()) with torch_mlir_module.context: - pm = PassManager.parse('torchscript-to-linalg-on-tensors-backend-pipeline') + pm = PassManager.parse('torchscript-module-to-linalg-on-tensors-backend-pipeline') pm.run(torch_mlir_module) print("\n\nLOWERED MLIR") diff --git a/examples/torchscript_resnet18_e2e.py b/examples/torchscript_resnet18_e2e.py index 37dc21b6e..9cf492336 100644 --- a/examples/torchscript_resnet18_e2e.py +++ b/examples/torchscript_resnet18_e2e.py @@ -108,7 +108,7 @@ mb.import_module(recursivescriptmodule._c, class_annotator) backend = refbackend.RefBackendLinalgOnTensorsBackend() with mb.module.context: - pm = PassManager.parse('torchscript-to-linalg-on-tensors-backend-pipeline') + pm = PassManager.parse('torchscript-module-to-linalg-on-tensors-backend-pipeline') pm.run(mb.module) compiled = backend.compile(mb.module)