46 lines
2 KiB
TableGen
46 lines
2 KiB
TableGen
//===-- MyExtension.td - Transform dialect tutorial --------*- tablegen -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines Transform dialect extension operations used in the
|
|
// Chapter 4 of the Transform dialect tutorial.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef MY_EXTENSION
|
|
#define MY_EXTENSION
|
|
|
|
include "mlir/Dialect/Transform/IR/MatchInterfaces.td"
|
|
include "mlir/Dialect/Transform/IR/TransformDialect.td"
|
|
include "mlir/Dialect/Transform/IR/TransformInterfaces.td"
|
|
include "mlir/IR/OpBase.td"
|
|
include "mlir/Interfaces/SideEffectInterfaces.td"
|
|
|
|
// Define the new operation. By convention, prefix its name with `match`
|
|
// followed by the name of the dialect extension.
|
|
def HasOperandSatisfyingOp : TransformDialectOp<"match.my.has_operand_satisfying",
|
|
[DeclareOpInterfaceMethods<MemoryEffectsOpInterface>,
|
|
DeclareOpInterfaceMethods<TransformOpInterface>,
|
|
// Indicate that the operation implements MatchOpInterface in addition to
|
|
// the TransformOpInterface. This interface is only used as a tag at this
|
|
// point and has no methods that are mandatory to implement.
|
|
MatchOpInterface,
|
|
SingleBlockImplicitTerminator<"::mlir::transform::YieldOp">]> {
|
|
let summary = "Succeed if any of the operands matches all nested criteria";
|
|
let arguments = (ins TransformHandleTypeInterface:$op);
|
|
let results = (outs TransformParamTypeInterface:$position,
|
|
Variadic<Transform_AnyHandleOrParamType>:$results);
|
|
|
|
// Match operations can be arbitrarily complex, e.g., containing regions.
|
|
let regions = (region SizedRegion<1>:$body);
|
|
let hasVerifier = 1;
|
|
let assemblyFormat = [{
|
|
$op `:` functional-type($op, results) attr-dict-with-keyword $body
|
|
}];
|
|
}
|
|
|
|
#endif // MY_EXTENSION
|