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/test/c10_dispatch
History
Stella Laurenzo e5433e314f Add capture function arguments. 
* Adds at::Tensor -> MlirValue tracking.
* Adds conversions for tensor and scalar types to MLIR types.
* Adds npcomp C APIs for constructing custom types.
* Reworks pybind include so as to get Torch pybind helpers (needed to pass at::Tensor type from Python->C++).
 		2020-10-01 18:59:58 -07:00
..
get_registered_ops.py Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
lit.local.cfg Move tests around so we can have dedicated tests for the c10 dispatcher. 2020-09-24 18:28:06 -07:00
module_builder.py Add capture function arguments. 2020-10-01 18:59:58 -07:00