1//===- TestTransformDialectExtension.td --------------------*- tablegen -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file declares the operations that are injected into the Transform
10// dialect through the extension mechanism, as a test.
11//
12//===----------------------------------------------------------------------===//
13
14#ifndef MLIR_TESTTRANSFORMDIALECTEXTENSION_TD
15#define MLIR_TESTTRANSFORMDIALECTEXTENSION_TD
16
17include "mlir/IR/OpBase.td"
18include "mlir/Dialect/Transform/IR/TransformDialect.td"
19include "mlir/Dialect/Transform/IR/TransformEffects.td"
20include "mlir/Dialect/Transform/IR/TransformInterfaces.td"
21include "mlir/Dialect/PDL/IR/PDLTypes.td"
22
23def TestProduceParamOrForwardOperandOp
24  : Op<Transform_Dialect, "test_produce_param_or_forward_operand",
25       [DeclareOpInterfaceMethods<TransformOpInterface>]> {
26  let arguments = (ins
27     Arg<Optional<PDL_Operation>, "", [TransformMappingRead]>:$operand,
28     OptionalAttr<I64Attr>:$parameter);
29  let results = (outs
30     Res<PDL_Operation, "",
31         [TransformMappingAlloc, TransformMappingWrite]>:$res);
32  let assemblyFormat = "(`from` $operand^)? ($parameter^)? attr-dict";
33  let cppNamespace = "::mlir::test";
34  let hasVerifier = 1;
35}
36
37def TestConsumeOperand : Op<Transform_Dialect, "test_consume_operand",
38     [DeclareOpInterfaceMethods<TransformOpInterface>]> {
39  let arguments = (ins
40    Arg<PDL_Operation, "",
41        [TransformMappingRead, TransformMappingFree]>:$operand);
42  let assemblyFormat = "$operand attr-dict";
43  let cppNamespace = "::mlir::test";
44}
45
46def TestConsumeOperandIfMatchesParamOrFail
47  : Op<Transform_Dialect, "test_consume_operand_if_matches_param_or_fail",
48       [DeclareOpInterfaceMethods<TransformOpInterface>]> {
49  let arguments = (ins
50    Arg<PDL_Operation, "",
51        [TransformMappingWrite, TransformMappingFree]>:$operand,
52    I64Attr:$parameter);
53  let assemblyFormat = "$operand `[` $parameter `]` attr-dict";
54  let cppNamespace = "::mlir::test";
55}
56
57def TestPrintRemarkAtOperandOp
58  : Op<Transform_Dialect, "test_print_remark_at_operand",
59       [DeclareOpInterfaceMethods<TransformOpInterface>]> {
60  let arguments = (ins
61    Arg<PDL_Operation, "",
62        [TransformMappingRead, PayloadIRRead]>:$operand,
63    StrAttr:$message);
64  let assemblyFormat = "$operand `,` $message attr-dict";
65  let cppNamespace = "::mlir::test";
66}
67
68def TestAddTestExtensionOp
69  : Op<Transform_Dialect, "test_add_test_extension",
70       [DeclareOpInterfaceMethods<TransformOpInterface>,
71        NoSideEffect]> {
72  let arguments = (ins StrAttr:$message);
73  let assemblyFormat = "$message attr-dict";
74  let cppNamespace = "::mlir::test";
75}
76
77def TestCheckIfTestExtensionPresentOp
78  : Op<Transform_Dialect, "test_check_if_test_extension_present",
79       [DeclareOpInterfaceMethods<TransformOpInterface>]> {
80  let arguments = (ins
81    Arg<PDL_Operation, "", [TransformMappingRead, PayloadIRRead]>:$operand);
82  let assemblyFormat = "$operand attr-dict";
83  let cppNamespace = "::mlir::test";
84}
85
86def TestRemapOperandPayloadToSelfOp
87  : Op<Transform_Dialect, "test_remap_operand_to_self",
88       [DeclareOpInterfaceMethods<TransformOpInterface>]> {
89  let arguments = (ins
90    Arg<PDL_Operation, "",
91        [TransformMappingRead, TransformMappingWrite, PayloadIRRead]>:$operand);
92  let assemblyFormat = "$operand attr-dict";
93  let cppNamespace = "::mlir::test";
94}
95
96def TestRemoveTestExtensionOp
97  : Op<Transform_Dialect, "test_remove_test_extension",
98       [DeclareOpInterfaceMethods<TransformOpInterface>,
99        NoSideEffect]> {
100  let assemblyFormat = "attr-dict";
101  let cppNamespace = "::mlir::test";
102}
103
104def TestTransformOpWithRegions
105  : Op<Transform_Dialect, "test_transform_op_with_regions",
106       [DeclareOpInterfaceMethods<TransformOpInterface>,
107        DeclareOpInterfaceMethods<MemoryEffectsOpInterface>]> {
108  let regions = (region AnyRegion:$first, AnyRegion:$second);
109  let assemblyFormat = "attr-dict-with-keyword regions";
110  let cppNamespace = "::mlir::test";
111}
112
113def TestBranchingTransformOpTerminator
114  : Op<Transform_Dialect, "test_branching_transform_op_terminator",
115       [Terminator, DeclareOpInterfaceMethods<TransformOpInterface>,
116        DeclareOpInterfaceMethods<MemoryEffectsOpInterface>]> {
117  let successors = (successor VariadicSuccessor<AnySuccessor>:$succ);
118  let cppNamespace = "::mlir::test";
119}
120
121#endif // MLIR_TESTTRANSFORMDIALECTEXTENSION_TD
122