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,
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,
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...); \
104 HANDLE(InterleaveSequencesOp, Unhandled);
106 HANDLE(RandomNumberInRangeOp, Unhandled);
114 HANDLE(SetCartesianProductOp, Unhandled);
146template <
typename ConcreteType,
typename ResultType = void,
147 typename... ExtraArgs>
151 auto *thisCast =
static_cast<ConcreteType *
>(
this);
152 return TypeSwitch<Type, ResultType>(type)
153 .template Case<ImmediateType, SequenceType, SetType, BagType, DictType,
154 LabelType, IndexType, IntegerType>(
155 [&](
auto expr) -> ResultType {
156 return thisCast->visitType(expr, args...);
158 .
template Case<ContextResourceTypeInterface>(
159 [&](
auto expr) -> ResultType {
160 return thisCast->visitContextResourceType(expr, args...);
162 .Default([&](
auto expr) -> ResultType {
163 if (&type.getDialect() ==
164 type.getContext()->getLoadedDialect<RTGDialect>())
167 return thisCast->visitExternalType(type, args...);
174 llvm::errs() <<
"Unknown RTG type: " << type <<
"\n";
191#define HANDLE(TYPETYPE, TYPEKIND) \
192 ResultType visitType(TYPETYPE type, ExtraArgs... args) { \
193 return static_cast<ConcreteType *>(this)->visit##TYPEKIND##Type(type, \
This helps visit TypeOp nodes.
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(ConstantOp, Unhandled)
HANDLE(SetUnionOp, 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.