torch-mlir/frontends/pytorch/csrc/c10_dispatch
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
..
CMakeLists.txt Add capture function arguments. 2020-10-01 18:59:58 -07:00
acap_dispatch.cpp Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
acap_dispatch.h Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
func_builder.cpp Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
func_builder.h Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
module_builder.cpp Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00
module_builder.h Add capture function arguments. 2020-10-01 18:59:58 -07:00
python_bindings.cpp Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00