tree 26e246add2654106cc836cb2abe290d6393e555b
parent ba4e03e4144915ddf10f0b5f7fec65b1674bd6b8
author Artem Kroviakov <71938912+akroviakov@users.noreply.github.com> 1757536406 +0200
committer Copybara-Service <copybara-worker@google.com> 1757536517 -0700

[MLIR][Vector] Add warp distribution for `scf.if` (#157119)

This PR adds `scf.if` op distribution to the existing `VectorDistribute`
patterns. The logic mostly follows that of `scf.for`: move op outside, wrap each
branch with `gpu.warp_execute_on_lane_0`. A notable difference to `scf.for` is
that each branch has its own set of escaping values, and `scf.if` itself does not
have block arguments.

GitOrigin-RevId: 7f007b572d63fe802a1f5587c74aae437f9f50e6