[all-commits] [llvm/llvm-project] 186c3d: [MLIR][Affine] Fix generateUnrolledLoop utility

[akshaybaviskar via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 186c3d6315f01669a46413df49d98366e33aa208
      https://github.com/llvm/llvm-project/commit/186c3d6315f01669a46413df49d98366e33aa208
  Author: Akshay Baviskar <akshay at polymagelabs.com>
  Date:   2023-04-06 (Thu, 06 Apr 2023)

  Changed paths:
    M mlir/lib/Dialect/Affine/Utils/LoopUtils.cpp
    M mlir/test/Dialect/SCF/loop-unroll.mlir

  Log Message:
  -----------
  [MLIR][Affine] Fix generateUnrolledLoop utility

generateUnrolledLoop was assuming that the yielded value is always generated in
the Block corresponding to the loop being unrolled. Thus, it was updating the
last yielded values with it's cloned value. However, if the yielded value is not
generated in the same Block then the cloned value and it's corresponding mapping
won't exist, resulting in a crash. Fix this.

Reviewed By: bondhugula

Differential Revision: https://reviews.llvm.org/D146931