Skip to content

Conversation

@wrongtest-intellif
Copy link
Contributor

Add a simple optimization to block form after compute-at when the result loop domain and extra loop predicate could be merged.
eg,

for i in range(87):
    T.where(i < 85): ...

Could be generated as

for i in range(85): ...

@ysh329 ysh329 changed the title [Compute-at] Make compute-ated block simple when the predicate could be merged [tir][Compute-at] Make compute-ated block simple when the predicate could be merged Apr 28, 2024
@wrongtest-intellif wrongtest-intellif force-pushed the add_compute_add_dom_simplify branch from c31f8c4 to f8c2132 Compare April 29, 2024 01:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants