mirror of https://github.com/llvm/torch-mlir
0a7ba62438
PyTorch allows new operators to be registered dynamically in modules. Torch-mlir already makes it fairly straightforward to add support for new operators, and this commit just extends that support to allow new PyTorch ops to come from a external module. This does *not* allow ops to be dynamically loaded into torch-mlir. Torch-mlir must still be compiled with support built-in. Add a `_torch_mlir_custom_op_example` subpackage to `torch_mlir` which registers an demonstration op. It will not be imported by default when importing torch_mlir. It's strictly for testing and documentation. Adds an end-to-end test for the `torch_mlir_custom_op_example::identity` op. With all these changes, we should now be actively testing PyTorch extension support with all future patches. |
||
|---|---|---|
| .. | ||
| TorchToLinalg | ||
| TorchToSCF | ||
| TorchToStd | ||
| TorchToTMTensor | ||
| TorchToTosa | ||
| Utils | ||
| CMakeLists.txt | ||
| PassDetail.h | ||
| Passes.cpp | ||