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
dialects: (dmp) A bunch of fixes in the lowering #1061
Conversation
Codecov ReportPatch coverage:
Additional details and impacted files@@ Coverage Diff @@
## main #1061 +/- ##
==========================================
- Coverage 86.92% 86.90% -0.02%
==========================================
Files 129 130 +1
Lines 19703 20452 +749
Branches 2981 3112 +131
==========================================
+ Hits 17126 17774 +648
- Misses 2070 2169 +99
- Partials 507 509 +2
☔ View full report in Codecov by Sentry. |
@@ -211,7 +214,7 @@ def _emit_memref_counts( | |||
if not all(dim.value.data >= 0 for dim in ssa_val.typ.shape.data): | |||
raise RuntimeError("MPI lowering does not support unknown-size memrefs!") | |||
|
|||
size = sum(dim.value.data for dim in ssa_val.typ.shape.data) | |||
size = prod(dim.value.data for dim in ssa_val.typ.shape.data) |
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.
seems like a useful helper to add on the type, potentially?
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.
Absolutely!
cst0 = arith.Constant.from_int_and_width(0, idx) | ||
cst1 = arith.Constant.from_int_and_width(1, idx) | ||
|
||
@Builder.implicit_region([idx, idx]) |
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.
yay! clients!
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.
It's so wonderful to use! Thanks for adding it!
No description provided.