[all-commits] [llvm/llvm-project] f7137d: [mlir][linalg] Fix dim(iter_arg) canonicalization

[Matthias Springer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f7137da174a43dd5460b72a21fbcaac7b7b74d7a
      https://github.com/llvm/llvm-project/commit/f7137da174a43dd5460b72a21fbcaac7b7b74d7a
  Author: Matthias Springer <springerm at google.com>
  Date:   2021-09-09 (Thu, 09 Sep 2021)

  Changed paths:
    M mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
    M mlir/test/Dialect/Linalg/canonicalize.mlir

  Log Message:
  -----------
  [mlir][linalg] Fix dim(iter_arg) canonicalization

Run a small analysis to see if the runtime type of the iter_arg is changing. Fold only if the runtime type stays the same. (Same as `DimOfIterArgFolder` in SCF.)

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