[all-commits] [llvm/llvm-project] 6c57b0: [mlir] improve and test TransformState::Extension

[ftynse via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 6c57b0debedaa5f211a39a8d6765ad4c74db4059
      https://github.com/llvm/llvm-project/commit/6c57b0debedaa5f211a39a8d6765ad4c74db4059
  Author: Alex Zinenko <zinenko at google.com>
  Date:   2022-05-03 (Tue, 03 May 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Transform/IR/TransformInterfaces.h
    M mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
    A mlir/test/Dialect/Transform/transform-state-extension.mlir
    M mlir/test/lib/Dialect/Transform/TestTransformDialectExtension.cpp
    M mlir/test/lib/Dialect/Transform/TestTransformDialectExtension.td
    A mlir/test/lib/Dialect/Transform/TestTransformStateExtension.h

  Log Message:
  -----------
  [mlir] improve and test TransformState::Extension

Add the mechanism for TransformState extensions to update the mapping between
Transform IR values and Payload IR operations held by the state. The mechanism
is intentionally restrictive, similarly to how results of the transform op are
handled.

Introduce test ops that exercise a simple extension that maintains information
across the application of multiple transform ops.

Reviewed By: nicolasvasilache

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