torch-mlir/frontends/pytorch/csrc
Stella Laurenzo ad3ddb9edb Implement torch.kernel_call capture.
* Had to stop short of modifying the function return signature because of a missing C-API upstream.
* Committing here is good enough for a test and will resolve the various TODOs about upstream APIs next.
2020-10-06 21:54:28 -07:00
..
c10_dispatch Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
type_dispatch Add capture function arguments. 2020-10-01 18:59:58 -07:00
CMakeLists.txt Add initial python bindings for c10 dispatcher internals. (#55) 2020-09-24 16:26:29 -07:00
init_python_bindings.cpp Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
init_python_bindings.h Add capture function arguments. 2020-10-01 18:59:58 -07:00
pybind.h Add capture function arguments. 2020-10-01 18:59:58 -07:00