torch-mlir

Commit Graph

Author	SHA1	Message	Date
Stella Laurenzo	a74a98094b	Add a new python script to auto-generate ATen op ODS definitions. (#43 ) * Add a new python script to auto-generate ATen op ODS definitions. * There is still some work on some of the ops to annotate correct types. * The ODS is not actually included into the dialect yet, but I'd like to commit it so that we can track changes. * Will reconcile this with the ops produced by the existing script in a followup. Still need to do some more iteration to reach parity.	2020-09-16 16:21:24 -07:00
Stella Laurenzo	d1ed6d260e	Initial work on a torch op registry. * This extracts metadata from python invocations (nearly) sufficient to generate ODS and a Torch IR translation table for most of the ops. * It also has the side effect of creating a data structure with meaningfully runnable examples suitable for an automated regression test. * There are some ops that are sufficiently complex/weird (like _convolution) that we'll just manually handle those. * See example output: https://gist.github.com/stellaraccident/60a58457b15e9184e224fa98a2658769	2020-08-28 15:20:55 -07:00

Author

SHA1

Message

Date

Stella Laurenzo

a74a98094b

Add a new python script to auto-generate ATen op ODS definitions. (#43 )

* Add a new python script to auto-generate ATen op ODS definitions.

* There is still some work on some of the ops to annotate correct types.
* The ODS is not actually included into the dialect yet, but I'd like to commit it so that we can track changes.
* Will reconcile this with the ops produced by the existing script in a followup. Still need to do some more iteration to reach parity.

2020-09-16 16:21:24 -07:00

Stella Laurenzo

d1ed6d260e

Initial work on a torch op registry.

* This extracts metadata from python invocations (nearly) sufficient to generate ODS and a Torch IR translation table for most of the ops.
* It also has the side effect of creating a data structure with meaningfully runnable examples suitable for an automated regression test.
* There are some ops that are sufficiently complex/weird (like _convolution) that we'll just manually handle those.
* See example output: https://gist.github.com/stellaraccident/60a58457b15e9184e224fa98a2658769

2020-08-28 15:20:55 -07:00

2 Commits (62738d3641ce3c7b10775b170b1287b84d6838c1)