[all-commits] [llvm/llvm-project] d26c42: [mlir][linalg] Control dimensions to pad.

[Tobias Gysi via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: d26c42af57fc23557202006e362ba6ba1e3e57dc
      https://github.com/llvm/llvm-project/commit/d26c42af57fc23557202006e362ba6ba1e3e57dc
  Author: gysit <gysit at google.com>
  Date:   2022-03-28 (Mon, 28 Mar 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/include/mlir/Dialect/Tensor/Utils/Utils.h
    M mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
    M mlir/lib/Dialect/Linalg/Utils/Utils.cpp
    M mlir/lib/Dialect/Tensor/Utils/Utils.cpp
    M mlir/test/Dialect/Linalg/codegen-strategy.mlir
    M mlir/test/Dialect/Linalg/pad.mlir
    M mlir/test/lib/Dialect/Linalg/TestLinalgCodegenStrategy.cpp

  Log Message:
  -----------
  [mlir][linalg] Control dimensions to pad.

This revision supports padding only a subset of the iteration dimensions via an additional padding-dimensions parameter. This control allows us to pad an operation in multiple steps. For example, one may want to pad only the output dimensions of a producer matmul fused into a consumer loop nest, before tiling and padding its reduction dimension.

Depends On D122309

Reviewed By: nicolasvasilache

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