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. |
||
|---|---|---|
| .. | ||
| python_deploy | ||
| torchscript_e2e_heavydep_tests | ||
| build_python_wheels.sh | ||
| build_standalone.sh | ||
| update_shape_lib.sh | ||
| update_torch_ods.sh | ||
| write_env_file.sh | ||