[all-commits] [llvm/llvm-project] 0e06ec: [mlir][linalg] Return tensor::PadOp handle from tr...

[Matthias Springer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0e06ec5961325c51ef2fa48a25c9b99038ff2f1b
      https://github.com/llvm/llvm-project/commit/0e06ec5961325c51ef2fa48a25c9b99038ff2f1b
  Author: Matthias Springer <me at m-sp.org>
  Date:   2023-07-04 (Tue, 04 Jul 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
    M mlir/lib/Dialect/Linalg/Transforms/Padding.cpp
    M mlir/python/mlir/dialects/_structured_transform_ops_ext.py
    A mlir/test/Dialect/Linalg/pad-to-specific-memory-space.mlir
    M mlir/test/Dialect/Linalg/transform-op-hoist-pad-build-packing-loop-nest.mlir
    M mlir/test/Dialect/Linalg/transform-op-hoist-pad.mlir
    M mlir/test/Dialect/Linalg/transform-op-pad.mlir
    M mlir/test/Dialect/Linalg/transform-ops-invalid.mlir
    M mlir/test/Dialect/Linalg/transform-ops.mlir

  Log Message:
  -----------
  [mlir][linalg] Return tensor::PadOp handle from transform op

"transform.structured.pad" now returns all `tensor::PadOp` in addition to the padded ops.

Also add a test case that shows how to force an allocation for "tensor.pad" ops with a custom memory space.

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