iree-org / iree-org/iree

Implement MaterializeEncoding for conv ops on GPU with IGEMM

Open
#17,606 0 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.