RTGVisitors.h

Go to the documentation of this file.

//===- RTGVisitors.h - RTG Dialect Visitors ---------------------*- C++ -*-===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// This file defines visitors that make it easier to work with RTG IR.
//
//===----------------------------------------------------------------------===//
 
#ifndef CIRCT_DIALECT_RTG_IR_RTGVISITORS_H
#define CIRCT_DIALECT_RTG_IR_RTGVISITORS_H
 
#include "circt/Dialect/RTG/IR/RTGISAAssemblyOpInterfaces.h"
#include "circt/Dialect/RTG/IR/RTGOpInterfaces.h"
#include "circt/Dialect/RTG/IR/RTGOps.h"
#include "circt/Dialect/RTG/IR/RTGTypeInterfaces.h"
#include "circt/Dialect/RTG/IR/RTGTypes.h"
#include "llvm/ADT/TypeSwitch.h"
 
namespace circt {
namespace rtg {
 
/// This helps visit TypeOp nodes.
template <typename ConcreteType, typename ResultType = void,
          typename... ExtraArgs>
class RTGOpVisitor {
public:
  ResultType dispatchOpVisitor(Operation *op, ExtraArgs... args) {
    auto *thisCast = static_cast<ConcreteType *>(this);
    return TypeSwitch<Operation *, ResultType>(op)
        .template Case<
            // Bags
            BagCreateOp, BagSelectRandomOp, BagDifferenceOp, BagUnionOp,
            BagUniqueSizeOp,
            // Labels
            LabelDeclOp, LabelUniqueDeclOp, LabelOp,
            // Registers
            FixedRegisterOp, VirtualRegisterOp,
            // RTG tests
            TestOp, TargetOp, YieldOp,
            // Sequences
            SequenceOp, SequenceClosureOp, InvokeSequenceOp,
            // Sets
            SetCreateOp, SetSelectRandomOp, SetDifferenceOp, SetUnionOp,
            SetSizeOp>([&](auto expr) -> ResultType {
          return thisCast->visitOp(expr, args...);
        })
        .template Case<ContextResourceOpInterface>(
            [&](auto expr) -> ResultType {
              return thisCast->visitContextResourceOp(expr, args...);
            })
        .Default([&](auto expr) -> ResultType {
          if (op->getDialect() ==
              op->getContext()->getLoadedDialect<RTGDialect>())
            return visitInvalidTypeOp(op, args...);
 
          return thisCast->visitExternalOp(op, args...);
        });
  }
 
  /// This callback is invoked on any RTG operations not handled properly by the
  /// TypeSwitch.
  ResultType visitInvalidTypeOp(Operation *op, ExtraArgs... args) {
    op->emitOpError("Unknown RTG operation: ") << op->getName();
    abort();
  }
 
  /// This callback is invoked on any operations that are not
  /// handled by the concrete visitor.
  ResultType visitUnhandledOp(Operation *op, ExtraArgs... args);
 
  ResultType visitContextResourceOp(ContextResourceOpInterface op,
                                    ExtraArgs... args) {
    return static_cast<ConcreteType *>(this)->visitUnhandledOp(op, args...);
  }
 
  ResultType visitExternalOp(Operation *op, ExtraArgs... args) {
    return ResultType();
  }
 
#define HANDLE(OPTYPE, OPKIND)                                                 \
  ResultType visitOp(OPTYPE op, ExtraArgs... args) {                           \
    return static_cast<ConcreteType *>(this)->visit##OPKIND##Op(op, args...);  \
  }
 
  HANDLE(SequenceOp, Unhandled);
  HANDLE(SequenceClosureOp, Unhandled);
  HANDLE(InvokeSequenceOp, Unhandled);
  HANDLE(SetCreateOp, Unhandled);
  HANDLE(SetSelectRandomOp, Unhandled);
  HANDLE(SetDifferenceOp, Unhandled);
  HANDLE(SetUnionOp, Unhandled);
  HANDLE(SetSizeOp, Unhandled);
  HANDLE(BagCreateOp, Unhandled);
  HANDLE(BagSelectRandomOp, Unhandled);
  HANDLE(BagDifferenceOp, Unhandled);
  HANDLE(BagUnionOp, Unhandled);
  HANDLE(BagUniqueSizeOp, Unhandled);
  HANDLE(LabelDeclOp, Unhandled);
  HANDLE(LabelUniqueDeclOp, Unhandled);
  HANDLE(LabelOp, Unhandled);
  HANDLE(TestOp, Unhandled);
  HANDLE(TargetOp, Unhandled);
  HANDLE(YieldOp, Unhandled);
  HANDLE(FixedRegisterOp, Unhandled);
  HANDLE(VirtualRegisterOp, Unhandled);
#undef HANDLE
};
 
/// This helps visit TypeOp nodes.
template <typename ConcreteType, typename ResultType = void,
          typename... ExtraArgs>
class RTGTypeVisitor {
public:
  ResultType dispatchTypeVisitor(Type type, ExtraArgs... args) {
    auto *thisCast = static_cast<ConcreteType *>(this);
    return TypeSwitch<Type, ResultType>(type)
        .template Case<SequenceType, SetType, BagType, DictType, LabelType,
                       IndexType, IntegerType>([&](auto expr) -> ResultType {
          return thisCast->visitType(expr, args...);
        })
        .template Case<ContextResourceTypeInterface>(
            [&](auto expr) -> ResultType {
              return thisCast->visitContextResourceType(expr, args...);
            })
        .Default([&](auto expr) -> ResultType {
          if (&type.getDialect() ==
              type.getContext()->getLoadedDialect<RTGDialect>())
            return visitInvalidType(type, args...);
 
          return thisCast->visitExternalType(type, args...);
        });
  }
 
  /// This callback is invoked on any RTG types not handled properly by the
  /// TypeSwitch.
  ResultType visitInvalidType(Type type, ExtraArgs... args) {
    llvm::errs() << "Unknown RTG type: " << type << "\n";
    abort();
  }
 
  /// This callback is invoked on any types that are not
  /// handled by the concrete visitor.
  ResultType visitUnhandledType(Type type, ExtraArgs... args);
 
  ResultType visitContextResourceType(ContextResourceTypeInterface type,
                                      ExtraArgs... args) {
    return static_cast<ConcreteType *>(this)->visitUnhandledType(type, args...);
  }
 
  ResultType visitExternalType(Type type, ExtraArgs... args) {
    return ResultType();
  }
 
#define HANDLE(TYPETYPE, TYPEKIND)                                             \
  ResultType visitType(TYPETYPE type, ExtraArgs... args) {                     \
    return static_cast<ConcreteType *>(this)->visit##TYPEKIND##Type(type,      \
                                                                    args...);  \
  }
 
  HANDLE(SequenceType, Unhandled);
  HANDLE(SetType, Unhandled);
  HANDLE(BagType, Unhandled);
  HANDLE(DictType, Unhandled);
  HANDLE(IndexType, Unhandled);
  HANDLE(IntegerType, Unhandled);
  HANDLE(LabelType, Unhandled);
#undef HANDLE
};
 
} // namespace rtg
} // namespace circt
 
#endif // CIRCT_DIALECT_RTG_IR_RTGVISITORS_H