tree 68b328e4a50995b1d27f125949fbd98208287387
parent 7de3b7a2b915b0ddfa7510c361a01be4912324d0
author Stella Laurenzo <laurenzo@google.com> 1590799307 -0700
committer Copybara-Service <copybara-worker@google.com> 1590799438 -0700

Upgrade to the IdentifyDispatchRegions2 pass.
* Has a nasty workaround for #2050, which was exposed by different DR formation.
* Removed the special case output fusion in favor of relying on optimization passes to do the right thing (which is necessary for dynamic shapes anyway). FoldCompatibleDispatchRegions is working reasonably for this, and giving it smaller atoms, in theory, makes more advanced optimization passes more possible).
* Added in the special cases for our current list of ops that don't support fusion.
* Added a special case to assign a lower benefit to ops that are known to materialize to a copy, increasing the chance that they fuse to ops which are doing useful work.
* Enabled the dynamic shape multi-layer-perceptron tests on VMLA. Tried enabling on vulkan but ran into an issue. Filed #2057

PiperOrigin-RevId: 313881073