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

Update torch_mlir_tensor.py 

typo
pull/843/head snapshot-20220508.438
yuhao 2022-05-07 22:09:42 +08:00 committed by Yi Zhang
parent 28be6511d2
commit 2e6a9c084e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/torch_mlir/eager_mode/torch_mlir_tensor.py
View File

@ -15,7 +15,7 @@ from torch_mlir_e2e_test.linalg_on_tensors_backends import refbackend
class TorchMLIRTensor(torch.Tensor):
"""Wrap torch.Tensor in orer to dispatch through torch-mlir instead of aten.
"""Wrap torch.Tensor in order to dispatch through torch-mlir instead of aten.
This class uses the _make_wrapper_subclass pattern to override __torch_dispatch__
in order to dispatch through torch-mlir instead of aten. Here we basically only unwrap and wrap