torch-mlir/lib/CAPI
Stella Laurenzo 029815152e Add remaining pieces to capture full example models.
* Adds Basicpy List, Tuple, Dict types and plumbs through C API.
* Started debugging the issues around aten::conv2d capture, but a PyTorch bug is suspected.
* Was able to manually verify that the basic conv2d forward test captures correctly with a workaround.
* Need to resolve some printing issues upstream and move these tests to an integration test target (they take ~seconds to run).
2020-10-19 22:16:59 -07:00
..
CMakeLists.txt Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
InitLLVM.cpp Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
Registration.cpp Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
Types.cpp Add remaining pieces to capture full example models. 2020-10-19 22:16:59 -07:00