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

[LLVMGPU] Generalize VectorContractOpInfo based on indexing maps #17625

Closed Groverkss closed 2 weeks ago

Groverkss commented 3 weeks ago

This patch generalizes VectorContractOpInfo to work on any kind of vector.contract. The "kind" field was not being used by any pass anyway and they were relying on m, n, k dims.

Fixes: https://github.com/iree-org/iree/issues/17585

raikonenfnu commented 2 weeks ago

Comment for documentation, this should fix: https://github.com/iree-org/iree/issues/17585 :)