torch-mlir

Commit Graph

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

Author

SHA1

Message

Date

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

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

2 Commits (30cfc6499f1fa937e558112715e7b51f39683150)