search

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.47k stars 551 forks source link

`EncodingAttr` to track type of source op #17756

Open lialan opened 3 days ago

lialan commented 3 days ago

This is to get encoding ready for ops other than MATMUL.

Resolves https://github.com/iree-org/iree/issues/17604

hanhanW commented 3 days ago

Cant we use the indexing maps to deduce that this is a matmul?

I think it is possible. We can use linalg::inferContractionDims(maps) to see if it is a contraction op.