[all-commits] [llvm/llvm-project] 35df2f: Refactor GenericPadTensorOpVectorizationPattern

[Yi Zhang via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 35df2f6fbd1ae2e6f9313454e5446212fcbcf90a
      https://github.com/llvm/llvm-project/commit/35df2f6fbd1ae2e6f9313454e5446212fcbcf90a
  Author: Yi Zhang <cathyzhyi at google.com>
  Date:   2021-07-07 (Wed, 07 Jul 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/lib/Dialect/Linalg/Transforms/Bufferize.cpp
    M mlir/lib/Dialect/Linalg/Transforms/Transforms.cpp
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    A mlir/test/Dialect/Linalg/generalize-pad-tensor.mlir
    A mlir/test/Integration/Dialect/Linalg/CPU/test-padtensor.mlir
    M mlir/test/lib/Dialect/Linalg/TestLinalgTransforms.cpp

  Log Message:
  -----------
  Refactor GenericPadTensorOpVectorizationPattern

Refactor the original code to rewrite a PadTensorOp into a
sequence of InitTensorOp, FillOp and InsertSliceOp without
vectorization by default. `GenericPadTensorOpVectorizationPattern`
provides a customized OptimizeCopyFn to vectorize the
copying step.

Reviewed By: silvas, nicolasvasilache, springerm

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