llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update to new name of torchscript-module-to-linalg-on-tensors-backend-pipeline

pull/345/head
Sean Silva 2021-09-30 19:45:43 +00:00
parent 89225b0cd8
commit bc62a7fbf3
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/resnet_inference.ipynb
View File

@ -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",

2
examples/torchfx/examples/example_add_tanh_sigmoid.py
View File

@ -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")

2
examples/torchscript_resnet18_e2e.py
View File

@ -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)