-
Notifications
You must be signed in to change notification settings - Fork 30
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[LowerToAIE] Handle linalg ops like other ops in switch statement #693
Conversation
@@ -396,9 +384,6 @@ LogicalResult coreToAIE(IRRewriter &rewriter, AMDAIE::CoreOp coreOp, | |||
return coreReleaseOpToAIE(rewriter, releaseOp, mapper, | |||
toBeErased); | |||
}) | |||
.Case<linalg::LinalgOp>([&](auto linalgOp) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just reading the description and PR, this seems like it is dropping general handling of LinalgOp
. Is that to mean that we dont expect LinalgOp
s to survive till here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Instead of handling them as a special case:
iree-amd-aie/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIELowerToAIE.cpp
Line 399 in 57cd0bc
.Case<linalg::LinalgOp>([&](auto linalgOp) { |
They are now handled like other not-special ops:
iree-amd-aie/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIELowerToAIE.cpp
Line 411 in 57cd0bc
.Default([&](Operation *op) { |
We (I) do expect LinalgOps to survive til here.
I thought this was a fix for the non-deterministic crash I'm seeing in this pass with convolution, but it's not. I think I've found the actual problem (IRMapping not being updated when operations are erased) I'll post a fix for this tomorrow |
really? where? |
|
4fbbf74
to
c4f4afe
Compare
Bundled with #705 |
This is (now) just a code simplification. Linalg Ops don't need special handling.