Giters

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Home Page:http://iree.dev/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implement MaterializeEncoding for conv ops on GPU with IGEMM

bjacob opened this issue · comments

commented

Once #17605 is implemented, we can start making use for these encodings to materialize for certain conv implementation strategies. In particular, on GPU targets where we want to use the IGEMM strategy, this benefits from the C dimension being innermost, so we can materialize the encoding there are a transposition (or a special case of pack op) to achieve that.