torch-mlir/frontends/pytorch/csrc/c10_dispatch
Stella Laurenzo ba03ecc652 Add public API for constructing a module/function to capture PyTorch ops.
* Uses the MLIR-C API since that will save us a lot of grief down the road (i.e. will give PyTorch and libMLIR/libNPCOMP the ability to skew version-wise).
* Quite a few TODOs and not yet populating the function in any way.
2020-09-29 14:23:22 -07:00
..
CMakeLists.txt Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
acap_dispatch.cpp Add boilerplate to do device capture (pytorch 1.6). 2020-09-28 10:30:54 -07:00
acap_dispatch.h Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
module_builder.cpp Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
module_builder.h Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
python_bindings.cpp Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00