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	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

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

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 (58b6033537a518758d6d3364da2d0a28df882feb)