1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
// RUN: mlir-tblgen -gen-op-defs -I %S/../../include %s | FileCheck %s
include "mlir/Interfaces/SideEffectInterfaces.td"
def TEST_Dialect : Dialect {
let name = "test";
}
class TEST_Op<string mnemonic, list<Trait> traits = []> :
Op<TEST_Dialect, mnemonic, traits>;
def CustomResource : Resource<"CustomResource">;
def SideEffectOpA : TEST_Op<"side_effect_op_a"> {
let arguments = (ins
Arg<Variadic<AnyMemRef>, "", [MemRead]>,
Arg<SymbolRefAttr, "", [MemRead]>:$symbol,
Arg<FlatSymbolRefAttr, "", [MemWrite]>:$flat_symbol,
Arg<OptionalAttr<SymbolRefAttr>, "", [MemRead]>:$optional_symbol
);
let results = (outs Res<AnyMemRef, "", [MemAlloc<CustomResource>]>);
}
def SideEffectOpB : TEST_Op<"side_effect_op_b",
[MemoryEffects<[MemWrite<CustomResource>]>]>;
// CHECK: void SideEffectOpA::getEffects
// CHECK: for (::mlir::Value value : getODSOperands(0))
// CHECK: effects.emplace_back(::mlir::MemoryEffects::Read::get(), value, ::mlir::SideEffects::DefaultResource::get());
// CHECK: effects.emplace_back(::mlir::MemoryEffects::Read::get(), getSymbolAttr(), ::mlir::SideEffects::DefaultResource::get());
// CHECK: effects.emplace_back(::mlir::MemoryEffects::Write::get(), getFlatSymbolAttr(), ::mlir::SideEffects::DefaultResource::get());
// CHECK: if (auto symbolRef = getOptionalSymbolAttr())
// CHECK: effects.emplace_back(::mlir::MemoryEffects::Read::get(), symbolRef, ::mlir::SideEffects::DefaultResource::get());
// CHECK: for (::mlir::Value value : getODSResults(0))
// CHECK: effects.emplace_back(::mlir::MemoryEffects::Allocate::get(), value, CustomResource::get());
// CHECK: void SideEffectOpB::getEffects
// CHECK: effects.emplace_back(::mlir::MemoryEffects::Write::get(), CustomResource::get());
|
