torch-mlir

Commit Graph

Author	SHA1	Message	Date
Aaron Arthurs	4fd9b4afb5	Import ATen conv2d conversion and test (#180 ) * Import ATen conv2d conversion and test This is a first attempt at expanding ATen-to-TCF conversion for the conv2d operator. Eventually, this will come in use when lowering a high-level conv-based model.	2021-03-12 17:21:16 -08:00
Sean Silva	1dfcfa9cd1	Add aten.mm op and "test" it e2e. Note that unlike aten.matmul which has dynamic behavior depending on the argument ranks (can do matrix-matrix, matrix-vector, batch matmul, etc.), aten.mm is just a vanilla matrix multiply, which can be lowered precisely to tcf.matmul. The "test" is really just an example that I stared at while getting my feet wet with this. We probably want something that actually tests this as part of `ninja check-npcomp`.	2020-11-20 17:21:24 -08:00
Stella Laurenzo	e60dc2470e	Add aten.maximum op and conversions from aten->tcf. * Conversions are very simple, suporting mul, maximum and add (alpha=1 only). * Example added with pass pipeline needed to run. * Much missing off of the golden path but sufficient for such simple cases.	2020-11-04 17:20:54 -08:00

Author

SHA1

Message

Date

Aaron Arthurs

4fd9b4afb5

Import ATen conv2d conversion and test (#180 )

* Import ATen conv2d conversion and test

This is a first attempt at expanding ATen-to-TCF conversion for the
conv2d operator. Eventually, this will come in use when lowering a
high-level conv-based model.

2021-03-12 17:21:16 -08:00

Sean Silva

1dfcfa9cd1

Add aten.mm op and "test" it e2e.

Note that unlike aten.matmul which has dynamic behavior
depending on the argument ranks (can do matrix-matrix, matrix-vector,
batch matmul, etc.), aten.mm is just a vanilla matrix
multiply, which can be lowered precisely to tcf.matmul.

The "test" is really just an example that I stared at while getting my
feet wet with this. We probably want something that actually tests this
as part of `ninja check-npcomp`.

2020-11-20 17:21:24 -08:00

Stella Laurenzo

e60dc2470e

Add aten.maximum op and conversions from aten->tcf.

* Conversions are very simple, suporting mul, maximum and add (alpha=1 only).
* Example added with pass pipeline needed to run.
* Much missing off of the golden path but sufficient for such simple cases.

2020-11-04 17:20:54 -08:00

3 Commits (4591884d0684a9c4a5a5fd030646cbdb145b79f6)