commit 42cdcd509663ac16cb96dc7eb0f7811c98ba0243
author Artem Kroviakov <71938912+akroviakov@users.noreply.github.com> Wed Sep 10 22:33:26 2025 +0200
committer Copybara-Service <copybara-worker@google.com> Wed Sep 10 13:35:17 2025 -0700
tree 26e246add2654106cc836cb2abe290d6393e555b
parent ba4e03e4144915ddf10f0b5f7fec65b1674bd6b8

[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