-
Notifications
You must be signed in to change notification settings - Fork 2
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
Unhack stencil.store result through bufferization #118
Conversation
c3b7786
to
a3e3002
Compare
a3e3002
to
5c15ce9
Compare
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.
Looks great from a first pass! Lemme know when it will not be a DRAFT!
@@ -1,4 +1,4 @@ | |||
// RUN: xdsl-opt -p "distribute-stencil{strategy=3d-grid slices=2,1,1 restrict_domain=false},canonicalize-dmp,convert-stencil-to-ll-mlir" %s | filecheck %s | |||
// RUN: xdsl-opt -p "canonicalize,cse,distribute-stencil{strategy=3d-grid slices=2,1,1 restrict_domain=false},shape-inference,canonicalize-dmp,stencil-bufferize,dmp-to-mpi{mpi_init=false},convert-stencil-to-ll-mlir,canonicalize,cse" %s | filecheck %s |
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.
This test is doing more than the original one; I think it is stronger, and this is good.
Is there a reason behind that?
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.
IIRC, canonicalization gets rid of some useless stencil generated operations, that make my bufferization ungracefully fail. (As-in, bufferization expects a canonicalized input)
It could arguably be handled differently but figured might as well add those generic optimizations everywhere; they help with the code and just remove work for heavier optimization passes!
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.
Thanks for the reply, can you add this (your reply's useful info) as a comment/info/doc in some place you consider it appropriate?
20025ee
to
51e6f97
Compare
Instead of using a hacked result on
stencil.store
, use astencil.store
and then restencil.load
.Use meanwhile added xDSL stuff and update pipelines and tests accordingly!
Filechecks have changed a lot since the pipeline is updated. I took the chance to add more xDSL canonicalization and CSE steps now that it's performing there too!