commit a0945cc1aafbe2c9eabc807ddb6e8397ec6d0716
author Max191 <44243577+Max191@users.noreply.github.com> Fri Aug 23 08:52:52 2024 -0700
committer GitHub <noreply@github.com> Fri Aug 23 11:52:52 2024 -0400
tree e0038ca17b561158f53b6be6e45838e7e3a55001
parent c6924b619d22276d5148ca5c608b986bdf51b217

[Flow] Add pass to bubble and hoist encoding ops out of dispatch regions (#18063)

This PR adds a new pass in the Flow data tiling pipeline to hoist
encoding ops out of their dispatch regions. After SetEncoding, the
encoding ops are inserted directly inside of the dispatch regions that
contain the data tiled ops. The set_encoding ops then need to be hoisted
out of the dispatch region in order to fuse into the producer dispatch.

Sometimes there may be producer operations fused into the same dispatch
as the data tiled op, in which case the set_encoding ops will have
producers inside of the dispatch. In order to hoist the set_encoding op,
it needs to be bubbled up through these producer operations until it has
no producers inside of its dispatch. This pass supports bubbling of
set_encoding ops through bit extending ops and broadcasting ops.

After this pass, all set_encoding ops should be outside of dispatch
regions, and they need to be fused with their producers. Another pass
will be added in the next PR to fuse set_encoding ops into their
producer dispatch regions or wrap them in a new dispatch region.

---------

Signed-off-by: Max Dawkins <max.dawkins@gmail.com>