torch-mlir

Commit Graph

Author	SHA1	Message	Date
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
Stella Laurenzo	b5f010284f	Add boilerplate to do device capture (pytorch 1.6). * Uses the new dispatcher API. * Just prints to the console for the moment when an op is captured. * Executes the op through the existing implementation.	2020-09-28 10:30:54 -07:00

Author

SHA1

Message

Date

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

Stella Laurenzo

b5f010284f

Add boilerplate to do device capture (pytorch 1.6).

* Uses the new dispatcher API.
* Just prints to the console for the moment when an op is captured.
* Executes the op through the existing implementation.

2020-09-28 10:30:54 -07:00

2 Commits (3d74337be0443440402e2419ce07c9c0ee77baf5)