mirror of https://github.com/llvm/torch-mlir
Update to new name of torchscript-module-to-linalg-on-tensors-backend-pipeline
parent
89225b0cd8
commit
bc62a7fbf3
|
@ -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",
|
||||
|
|
|
@ -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")
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue