torch-mlir

Commit Graph

Author	SHA1	Message	Date
Phoenix Meadowlark	699bf5df45	Add cos_e2e.py, test_utils and support for tensor inputs (#134 )	2020-11-24 19:02:50 -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

Author

SHA1

Message

Date

Phoenix Meadowlark

699bf5df45

Add cos_e2e.py, test_utils and support for tensor inputs (#134 )

2020-11-24 19:02:50 -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

2 Commits (572d198b683bfcfd57e75a26e943340ccba03719)