[all-commits] [llvm/llvm-project] cbd802: [MLIR][Vector] Fix `scf.for` block-argument yields...

[Akimasa Watanuki via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: cbd802b07289c210d0f9f299d25176cd797be1b7
      https://github.com/llvm/llvm-project/commit/cbd802b07289c210d0f9f299d25176cd797be1b7
  Author: Akimasa Watanuki <mencotton0410 at gmail.com>
  Date:   2026-04-20 (Mon, 20 Apr 2026)

  Changed paths:
    M mlir/lib/Dialect/Vector/Transforms/VectorDistribute.cpp
    A mlir/test/Dialect/Vector/warp-distribute-scf-for-block-args.mlir

  Log Message:
  -----------
  [MLIR][Vector] Fix `scf.for` block-argument yields in warp distribution (#192247)

Teach WarpOpScfForOp to remap yielded `scf.for` body block arguments
through `argMapping` before creating the replacement `gpu.yield`.

Handle yielded loop-carried values and other `scf.for` body block
arguments when moving the loop body into the new inner warp op, instead
of reusing the pre-merge values.

Add a regression test for yielding a loop-carried block argument during
warp distribution.

Fix #186573



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications