iree-org / iree-org/wave

[water] Use InferIndexExprsOpInterface descriptions in ScaledMmaOp EPT propagation

Open
#1,189 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
59
Forks
32
PR merge metrics
No merged PRs in 30d

Description

In `ScaledMmaOp::propagateElementsPerThreadBackward`, we have a local `OperandInfo` struct that duplicates the operand descriptions already provided by `getIndexExprValuesAndDescriptions` on `InferIndexExprsOpInterface`.
We currently avoid the interface dependency and may entirely remove EPT at some point, so this is tracked as a future cleanup.
See the TODO in `water/lib/Dialect/Wave/IR/WaveOps.cpp` in `ScaledMmaOp::propagateElementsPerThreadBackward`.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.