torch-mlir/lib/CAPI
Stella Laurenzo 78a3c90758 Add TorchScript graph importer.
* Does not handle all features yet but should conservatively fail on unsupported things.
* Location tracking is still somewhat mismatched between what TorchScript and MLIR do. Likely need a better heuristic for tracking locations from defs for nodes that do not carry location.
* Sets the ground-work for a specialized/generic split but only implements the generic side.
* Had some evidence that this requires a recent bump of PT nightly (within the last month) to pick up pybind11 2.6, which includes some cross-module symbol fixes (vs the previously sync'd version). No source changes, but older versions fail to cast function types at runtime.
2020-11-23 14:20:09 -08: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 TorchScript graph importer. 2020-11-23 14:20:09 -08:00