13#ifndef CIRCT_DIALECT_RTG_IR_RTGVISITORS_H
14#define CIRCT_DIALECT_RTG_IR_RTGVISITORS_H
21#include "llvm/ADT/TypeSwitch.h"
27template <
typename ConcreteType,
typename ResultType = void,
28 typename... ExtraArgs>
32 auto *thisCast =
static_cast<ConcreteType *
>(
this);
33 return TypeSwitch<Operation *, ResultType>(op)
38 BagCreateOp, BagSelectRandomOp, BagDifferenceOp, BagUnionOp,
39 BagUniqueSizeOp, BagConvertToSetOp,
41 OnContextOp, ContextSwitchOp,
43 LabelDeclOp, LabelUniqueDeclOp, LabelOp,
45 FixedRegisterOp, VirtualRegisterOp,
47 TestOp, TargetOp, YieldOp, ValidateOp, TestSuccessOp, TestFailureOp,
49 RandomNumberInRangeOp,
51 SequenceOp, GetSequenceOp, SubstituteSequenceOp,
52 RandomizeSequenceOp, EmbedSequenceOp, InterleaveSequencesOp,
54 SetCreateOp, SetSelectRandomOp, SetDifferenceOp, SetUnionOp,
55 SetSizeOp, SetCartesianProductOp, SetConvertToBagOp,
57 ArrayCreateOp, ArrayExtractOp, ArrayInjectOp, ArraySizeOp,
59 TupleCreateOp, TupleExtractOp,
61 IntToImmediateOp, ConcatImmediateOp, SliceImmediateOp,
63 MemoryAllocOp, MemoryBaseAddressOp, MemorySizeOp,
67 CommentOp>([&](
auto expr) -> ResultType {
68 return thisCast->visitOp(expr, args...);
70 .Default([&](
auto expr) -> ResultType {
71 if (op->getDialect() ==
72 op->getContext()->getLoadedDialect<RTGDialect>())
75 return thisCast->visitExternalOp(op, args...);
82 op->emitOpError(
"Unknown RTG operation: ") << op->getName();
94#define HANDLE(OPTYPE, OPKIND) \
95 ResultType visitOp(OPTYPE op, ExtraArgs... args) { \
96 return static_cast<ConcreteType *>(this)->visit##OPKIND##Op(op, args...); \
94#define HANDLE(OPTYPE, OPKIND) \ …
104 HANDLE(InterleaveSequencesOp, Unhandled);
106 HANDLE(RandomNumberInRangeOp, Unhandled);
114 HANDLE(SetCartesianProductOp, Unhandled);
151template <
typename ConcreteType,
typename ResultType = void,
152 typename... ExtraArgs>
156 auto *thisCast =
static_cast<ConcreteType *
>(
this);
157 return TypeSwitch<Type, ResultType>(type)
158 .template Case<ImmediateType, SequenceType, SetType, BagType, DictType,
159 LabelType, IndexType, IntegerType>(
160 [&](
auto expr) -> ResultType {
161 return thisCast->visitType(expr, args...);
163 .
template Case<ContextResourceTypeInterface>(
164 [&](
auto expr) -> ResultType {
165 return thisCast->visitContextResourceType(expr, args...);
167 .Default([&](
auto expr) -> ResultType {
168 if (&type.getDialect() ==
169 type.getContext()->getLoadedDialect<RTGDialect>())
172 return thisCast->visitExternalType(type, args...);
179 llvm::errs() <<
"Unknown RTG type: " << type <<
"\n";
196#define HANDLE(TYPETYPE, TYPEKIND) \
197 ResultType visitType(TYPETYPE type, ExtraArgs... args) { \
198 return static_cast<ConcreteType *>(this)->visit##TYPEKIND##Type(type, \
This helps visit TypeOp nodes.
HANDLE(ConcatImmediateOp, Unhandled)
HANDLE(ValidateOp, Unhandled)
HANDLE(MemoryAllocOp, Unhandled)
HANDLE(YieldOp, Unhandled)
ResultType visitExternalOp(Operation *op, ExtraArgs... args)
HANDLE(TupleCreateOp, Unhandled)
HANDLE(MemorySizeOp, Unhandled)
HANDLE(MemoryBaseAddressOp, Unhandled)
HANDLE(TestOp, Unhandled)
HANDLE(ArrayInjectOp, Unhandled)
ResultType visitInvalidTypeOp(Operation *op, ExtraArgs... args)
This callback is invoked on any RTG operations not handled properly by the TypeSwitch.
HANDLE(BagUniqueSizeOp, Unhandled)
HANDLE(EmbedSequenceOp, Unhandled)
HANDLE(OnContextOp, Unhandled)
HANDLE(BagCreateOp, Unhandled)
HANDLE(LabelUniqueDeclOp, Unhandled)
HANDLE(TestSuccessOp, Unhandled)
HANDLE(ConstantOp, Unhandled)
HANDLE(TestFailureOp, Unhandled)
HANDLE(SetUnionOp, Unhandled)
HANDLE(SliceImmediateOp, Unhandled)
HANDLE(ContextSwitchOp, Unhandled)
HANDLE(RandomizeSequenceOp, Unhandled)
HANDLE(IntToImmediateOp, Unhandled)
HANDLE(BagConvertToSetOp, Unhandled)
HANDLE(VirtualRegisterOp, Unhandled)
HANDLE(SetConvertToBagOp, Unhandled)
HANDLE(BagDifferenceOp, Unhandled)
HANDLE(InterleaveSequencesOp, Unhandled)
HANDLE(FixedRegisterOp, Unhandled)
HANDLE(RandomNumberInRangeOp, Unhandled)
HANDLE(BagUnionOp, Unhandled)
HANDLE(SetCartesianProductOp, Unhandled)
HANDLE(SetDifferenceOp, Unhandled)
HANDLE(TargetOp, Unhandled)
HANDLE(SetSizeOp, Unhandled)
HANDLE(SetCreateOp, Unhandled)
HANDLE(SetSelectRandomOp, Unhandled)
HANDLE(BagSelectRandomOp, Unhandled)
ResultType dispatchOpVisitor(Operation *op, ExtraArgs... args)
HANDLE(CommentOp, Unhandled)
HANDLE(SubstituteSequenceOp, Unhandled)
HANDLE(LabelDeclOp, Unhandled)
ResultType visitUnhandledOp(Operation *op, ExtraArgs... args)
This callback is invoked on any operations that are not handled by the concrete visitor.
HANDLE(LabelOp, Unhandled)
HANDLE(MemoryBlockDeclareOp, Unhandled)
HANDLE(ArrayExtractOp, Unhandled)
HANDLE(ArrayCreateOp, Unhandled)
HANDLE(SequenceOp, Unhandled)
HANDLE(TupleExtractOp, Unhandled)
HANDLE(ArraySizeOp, Unhandled)
HANDLE(GetSequenceOp, Unhandled)
This helps visit TypeOp nodes.
HANDLE(BagType, Unhandled)
ResultType dispatchTypeVisitor(Type type, ExtraArgs... args)
HANDLE(IndexType, Unhandled)
HANDLE(DictType, Unhandled)
HANDLE(IntegerType, Unhandled)
HANDLE(SequenceType, Unhandled)
HANDLE(SetType, Unhandled)
HANDLE(LabelType, Unhandled)
ResultType visitExternalType(Type type, ExtraArgs... args)
ResultType visitContextResourceType(ContextResourceTypeInterface type, ExtraArgs... args)
HANDLE(ImmediateType, Unhandled)
ResultType visitInvalidType(Type type, ExtraArgs... args)
This callback is invoked on any RTG types not handled properly by the TypeSwitch.
ResultType visitUnhandledType(Type type, ExtraArgs... args)
This callback is invoked on any types that are not handled by the concrete visitor.
The InstanceGraph op interface, see InstanceGraphInterface.td for more details.