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

Fix typo in lazytensor example passes (#385)

pull/363/head
Ramiro Leal-Cavazos 2021-10-27 14:40:40 -05:00 committed by GitHub
parent 5009cbf55c
commit 029c30c060
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/lazytensor_tanh.py
View File

@ -52,7 +52,7 @@ cu = CompilationUnit()
func_name = 'my_method' func_name = 'my_method'
script_function = cu.create_function(func_name, graph) script_function = cu.create_function(func_name, graph)
# `build_module` takes he torch.jit.ScriptFunction and the # `build_module` takes the torch.jit.ScriptFunction and the
# annotation on the operand types, and outputs an `ir.Module` # annotation on the operand types, and outputs an `ir.Module`
# with a single function representing the ScriptFunction in # with a single function representing the ScriptFunction in
# the torch MLIR dialect # the torch MLIR dialect
@ -65,7 +65,7 @@ mlir_module.dump()
# Compile the torch MLIR and execute the compiled program # Compile the torch MLIR and execute the compiled program
with mlir_module.context: with mlir_module.context:
pm = PassManager.parse('torchscript-function-to-torch-backend-pipeline,torch-backend-to-linalg-on-tensors-backend-pipeline') pm = PassManager.parse('torch-function-to-torch-backend-pipeline,torch-backend-to-linalg-on-tensors-backend-pipeline')
pm.run(mlir_module) pm.run(mlir_module)
print("BEFORE LINALG-ON-TENSORS BACKEND PIPELINE") print("BEFORE LINALG-ON-TENSORS BACKEND PIPELINE")