Implement MaterializeEncoding for conv ops on GPU with IGEMM
- Dominant language
- C++
- Stars
- 3.9k
- Forks
- 1k
- Avg merge
- 4d 16h
- Merged PRs (30d)
- 47
Description
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.
Contributor guide
Research direction
Start by reviewing the prerequisite work in issue #17605 and then trace how convolution encodings are represented for GPU targets using the IGEMM strategy. Done means MaterializeEncoding supports these convolution operations and materializes the encoding so the C dimension is innermost, using a transpose or pack operation as appropriate.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- compilers, machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100