[all-commits] [llvm/llvm-project] 1a2bb0: [MLIR][LINALG] Add canonicalization pattern in `li...

[Prateek Gupta via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 1a2bb03edab9d7aa31beb587d0c863acc6715d27
      https://github.com/llvm/llvm-project/commit/1a2bb03edab9d7aa31beb587d0c863acc6715d27
  Author: Prateek Gupta <gprateek93 at gmail.com>
  Date:   2022-02-21 (Mon, 21 Feb 2022)

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

  Log Message:
  -----------
  [MLIR][LINALG] Add canonicalization pattern in `linalg.generic` op for static shape inference.

This commit adds canonicalization pattern in `linalg.generic` op
for static shape inference. If any of the inputs or outputs have
static shape or is casted from a tensor of static shape, then
shapes of all the inputs and outputs can be inferred by using the
affine map of the static shape input/output.

Signed-Off-By: Prateek Gupta <prateek at nod-labs.com>

Reviewed By: mravishankar

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