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

Add link to forum post describing `__torch_dispatch__`

pull/731/head snapshot-20220401.363
Sean Silva 2022-04-01 10:10:43 -07:00 committed by GitHub
parent 51d4d55f8a
commit 14cf87633c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -21,7 +21,9 @@ class TorchMLIRTensor(torch.Tensor):
in order to dispatch through torch-mlir instead of aten. Here we basically only unwrap and wrap in order to dispatch through torch-mlir instead of aten. Here we basically only unwrap and wrap
torch.Tensors. Most of the heavy lifting is done in the adjacent torch_mlir_dispatch module. torch.Tensors. Most of the heavy lifting is done in the adjacent torch_mlir_dispatch module.
More documentation on how this pattern works can be found in this RFC More documentation on how this pattern works can be found in this forum post
https://dev-discuss.pytorch.org/t/what-and-why-is-torch-dispatch/557
and this RFC
https://github.com/pytorch/rfcs/blob/master/RFC-0001-torch-function-for-methods.md#process-followed-during-a-functionmethod-call https://github.com/pytorch/rfcs/blob/master/RFC-0001-torch-function-for-methods.md#process-followed-during-a-functionmethod-call
and this repo with many examples and this repo with many examples
https://github.com/albanD/subclass_zoo https://github.com/albanD/subclass_zoo