torch-mlir

Commit Graph

Author	SHA1	Message	Date
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
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

Author

SHA1

Message

Date

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

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

2 Commits (d9dc16a9be5df3b1bfb89834cb8d85a45036fc52)