052d24af29
The side effect infrastructure is based on the Effect and Resource class templates, instances of instantiations of which are constructed as thread-local singletons. With this scheme, it is impossible to further parameterize either of those, or the EffectInstance class that contains pointers to an Effect and Resource instances. Such a parameterization is necessary to express more detailed side effects, e.g. those of a loop or a function call with affine operations inside where it is possible to precisely specify the slices of accessed buffers. Include an additional Attribute to EffectInstance class for further parameterization. This allows to leverage the dialect-specific registration and uniquing capabilities of the attribute infrastructure without requiring Effect or Resource instantiations to be attached to a dialect themselves. Split out the generic part of the side effect Tablegen classes into a separate file to avoid generating built-in MemoryEffect interfaces when processing any .td file that includes SideEffectInterfaceBase.td. Reviewed By: rriddle Differential Revision: https://reviews.llvm.org/D91493
71 lines
2.3 KiB
TableGen
71 lines
2.3 KiB
TableGen
//===-- TestInterfaces.td - Test dialect interfaces --------*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef MLIR_TEST_DIALECT_TEST_INTERFACES
|
|
#define MLIR_TEST_DIALECT_TEST_INTERFACES
|
|
|
|
include "mlir/IR/OpBase.td"
|
|
include "mlir/Interfaces/SideEffectInterfaceBase.td"
|
|
|
|
// A type interface used to test the ODS generation of type interfaces.
|
|
def TestTypeInterface : TypeInterface<"TestTypeInterface"> {
|
|
let methods = [
|
|
InterfaceMethod<"Prints the type name.",
|
|
"void", "printTypeA", (ins "Location":$loc), [{
|
|
emitRemark(loc) << $_type << " - TestA";
|
|
}]
|
|
>,
|
|
InterfaceMethod<"Prints the type name.",
|
|
"void", "printTypeB", (ins "Location":$loc),
|
|
[{}], /*defaultImplementation=*/[{
|
|
emitRemark(loc) << $_type << " - TestB";
|
|
}]
|
|
>,
|
|
InterfaceMethod<"Prints the type name.",
|
|
"void", "printTypeC", (ins "Location":$loc)
|
|
>,
|
|
// It should be possible to use the interface type name as result type
|
|
// as well as in the implementation.
|
|
InterfaceMethod<"Prints the type name and returns the type as interface.",
|
|
"TestTypeInterface", "printTypeRet", (ins "Location":$loc),
|
|
[{}], /*defaultImplementation=*/[{
|
|
emitRemark(loc) << $_type << " - TestRet";
|
|
return $_type;
|
|
}]
|
|
>,
|
|
];
|
|
let extraClassDeclaration = [{
|
|
/// Prints the type name.
|
|
void printTypeD(Location loc) const {
|
|
emitRemark(loc) << *this << " - TestD";
|
|
}
|
|
}];
|
|
let extraTraitClassDeclaration = [{
|
|
/// Prints the type name.
|
|
void printTypeE(Location loc) const {
|
|
emitRemark(loc) << $_type << " - TestE";
|
|
}
|
|
}];
|
|
}
|
|
|
|
def TestEffectOpInterface
|
|
: EffectOpInterfaceBase<"TestEffectOpInterface",
|
|
"::mlir::TestEffects::Effect"> {
|
|
let cppNamespace = "::mlir";
|
|
}
|
|
|
|
class TestEffect<string effectName>
|
|
: SideEffect<TestEffectOpInterface, effectName, DefaultResource>;
|
|
|
|
class TestEffects<list<TestEffect> effects = []>
|
|
: SideEffectsTraitBase<TestEffectOpInterface, effects>;
|
|
|
|
def TestConcreteEffect : TestEffect<"TestEffects::Concrete">;
|
|
|
|
#endif // MLIR_TEST_DIALECT_TEST_INTERFACES
|