xref: /llvm-project/mlir/test/mlir-tblgen/op-side-effects.td (revision 2c1ae801e1b66a09a15028ae4ba614e0911eec00) 
1// RUN: mlir-tblgen -gen-op-defs -I %S/../../include %s | FileCheck %s
2
3include "mlir/Interfaces/SideEffectInterfaces.td"
4
5def TEST_Dialect : Dialect {
6  let name = "test";
7}
8class TEST_Op<string mnemonic, list<Trait> traits = []> :
9    Op<TEST_Dialect, mnemonic, traits>;
10
11def CustomResource : Resource<"CustomResource">;
12
13def SideEffectOpA : TEST_Op<"side_effect_op_a"> {
14  let arguments = (ins
15    Arg<Variadic<AnyMemRef>, "", [MemRead]>,
16    Arg<AnyMemRef, "", [MemWriteAt<1, FullEffect>]>,
17    Arg<SymbolRefAttr, "", [MemRead]>:$symbol,
18    Arg<FlatSymbolRefAttr, "", [MemWrite]>:$flat_symbol,
19    Arg<OptionalAttr<SymbolRefAttr>, "", [MemRead]>:$optional_symbol
20  );
21  let results = (outs Res<AnyMemRef, "", [MemAlloc<CustomResource, 0>]>);
22}
23
24def SideEffectOpB : TEST_Op<"side_effect_op_b",
25    [MemoryEffects<[MemWrite<CustomResource, 0>]>]>;
26
27// CHECK: void SideEffectOpA::getEffects
28// CHECK:  {
29// CHECK:    auto valueRange = getODSOperandIndexAndLength(0);
30// CHECK:    for (unsigned idx = valueRange.first; idx < valueRange.first + valueRange.second; idx++) {
31// CHECK:      effects.emplace_back(::mlir::MemoryEffects::Read::get(), &getOperation()->getOpOperand(idx), 0, false, ::mlir::SideEffects::DefaultResource::get());
32// CHECK:    }
33// CHECK:  }
34// CHECK:  {
35// CHECK:    auto valueRange = getODSOperandIndexAndLength(1);
36// CHECK:    for (unsigned idx = valueRange.first; idx < valueRange.first + valueRange.second; idx++) {
37// CHECK:      effects.emplace_back(::mlir::MemoryEffects::Write::get(), &getOperation()->getOpOperand(idx), 1, true, ::mlir::SideEffects::DefaultResource::get());
38// CHECK:    }
39// CHECK:  }
40// CHECK:   effects.emplace_back(::mlir::MemoryEffects::Read::get(), getSymbolAttr(), 0, false, ::mlir::SideEffects::DefaultResource::get());
41// CHECK:   effects.emplace_back(::mlir::MemoryEffects::Write::get(), getFlatSymbolAttr(), 0, false, ::mlir::SideEffects::DefaultResource::get());
42// CHECK:   if (auto symbolRef = getOptionalSymbolAttr())
43// CHECK:     effects.emplace_back(::mlir::MemoryEffects::Read::get(), symbolRef, 0, false, ::mlir::SideEffects::DefaultResource::get());
44// CHECK:  {
45// CHECK:    auto valueRange = getODSResultIndexAndLength(0);
46// CHECK:    for (unsigned idx = valueRange.first; idx < valueRange.first + valueRange.second; idx++) {
47// CHECK:      effects.emplace_back(::mlir::MemoryEffects::Allocate::get(), getOperation()->getOpResult(idx), 0, false, CustomResource::get());
48// CHECK:    }
49// CHECK:  }
50
51// CHECK: void SideEffectOpB::getEffects
52// CHECK:   effects.emplace_back(::mlir::MemoryEffects::Write::get(), 0, false, CustomResource::get());
53