[all-commits] [llvm/llvm-project] c788ca: [mlir][linalg] Fix FoldConstantTranspose execution...

[Lei Zhang via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c788cad83b6b5c24f8160f9fc11a69dd7beafb8b
      https://github.com/llvm/llvm-project/commit/c788cad83b6b5c24f8160f9fc11a69dd7beafb8b
  Author: Lei Zhang <antiagainst at google.com>
  Date:   2021-10-28 (Thu, 28 Oct 2021)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp

  Log Message:
  -----------
  [mlir][linalg] Fix FoldConstantTranspose execution inefficiency

* Move SmallVectors outside of inner loops to avoid frequent
  allocations and deallocations
* Calculate linearized index and call flat range getters to
  avoid internal shape querying behind `getValue`.

Reviewed By: mravishankar

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