torch-mlir/python/npcomp
Stella Laurenzo d1ed6d260e Initial work on a torch op registry.
* This extracts metadata from python invocations (nearly) sufficient to generate ODS and a Torch IR translation table for most of the ops.
* It also has the side effect of creating a data structure with meaningfully runnable examples suitable for an automated regression test.
* There are some ops that are sufficiently complex/weird (like _convolution) that we'll just manually handle those.
* See example output: https://gist.github.com/stellaraccident/60a58457b15e9184e224fa98a2658769
2020-08-28 15:20:55 -07:00
..
compiler Wire through codegen and runtime dependencies. 2020-07-10 22:57:26 -07:00
dialect Some fixes to get npcomp building and passing on windows. 2020-07-01 21:28:04 -07:00
frontends Add pytorch interface to ATen Dialect (#30) 2020-08-21 11:22:47 -07:00
torch Initial work on a torch op registry. 2020-08-28 15:20:55 -07:00
tracing Remove use of namedtupled defaults kwarg. 2020-08-04 18:41:22 -07:00
utils Add script tools/format_source.sh and run it on all python and c++ sources. 2020-06-13 14:53:54 -07:00
__init__.py Cleanup python namespace a bit for standalone use. 2020-05-02 21:54:13 -07:00
decorators.py Add script tools/format_source.sh and run it on all python and c++ sources. 2020-06-13 14:53:54 -07:00
exporter.py Add script tools/format_source.sh and run it on all python and c++ sources. 2020-06-13 14:53:54 -07:00
smoketest.py Move python native library to python_native/_npcomp...so. 2020-05-06 22:44:12 -07:00
types.py Add script tools/format_source.sh and run it on all python and c++ sources. 2020-06-13 14:53:54 -07:00