llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/csrc/c10_dispatch
History
Stella Laurenzo 30cfc6499f Create public API for torch_mlir python code. 
* Adds a trampoline/loader 'torch_mlir' module.
* Plumbs through the MLIR python Context and Module creation, interoping with the MLIR Python API (resolves TODO on creating with own context and accessing the module being built).
* Inter-module Python API interop is still a bit rough but workable via the capsule mechanism. Can be evolved later.
* Exports the frontends/pytorch python sources to the project python/ build directory.
* Requires D89294 to land.
 		2020-10-13 16:36:49 -07:00
..
CMakeLists.txt Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
acap_dispatch.cpp Set PyTorch captured function return type. 2020-10-07 10:14:34 -07:00
acap_dispatch.h Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
func_builder.cpp Set PyTorch captured function return type. 2020-10-07 10:14:34 -07:00
func_builder.h Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
module_builder.cpp Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00
module_builder.h Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00
python_bindings.cpp Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00