65 lines · cpp
1//===- SideEffects.cpp - SideEffect classes -------------------------------===//2//3// Part of the MLIR Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "mlir/TableGen/SideEffects.h"10#include "llvm/ADT/Twine.h"11#include "llvm/TableGen/Record.h"12 13using namespace mlir;14using namespace mlir::tblgen;15 16//===----------------------------------------------------------------------===//17// SideEffect18//===----------------------------------------------------------------------===//19 20StringRef SideEffect::getName() const {21 return def->getValueAsString("effect");22}23 24StringRef SideEffect::getBaseEffectName() const {25 return def->getValueAsString("baseEffectName");26}27 28std::string SideEffect::getInterfaceTrait() const {29 StringRef trait = def->getValueAsString("interfaceTrait");30 StringRef cppNamespace = def->getValueAsString("cppNamespace");31 return cppNamespace.empty() ? trait.str()32 : (cppNamespace + "::" + trait).str();33}34 35StringRef SideEffect::getResource() const {36 return def->getValueAsString("resource");37}38 39int64_t SideEffect::getStage() const { return def->getValueAsInt("stage"); }40 41bool SideEffect::getEffectOnfullRegion() const {42 return def->getValueAsBit("effectOnFullRegion");43}44 45bool SideEffect::classof(const Operator::VariableDecorator *var) {46 return var->getDef().isSubClassOf("SideEffect");47}48 49//===----------------------------------------------------------------------===//50// SideEffectsTrait51//===----------------------------------------------------------------------===//52 53Operator::var_decorator_range SideEffectTrait::getEffects() const {54 auto *listInit = dyn_cast<llvm::ListInit>(def->getValueInit("effects"));55 return {listInit->begin(), listInit->end()};56}57 58StringRef SideEffectTrait::getBaseEffectName() const {59 return def->getValueAsString("baseEffectName");60}61 62bool SideEffectTrait::classof(const Trait *t) {63 return t->getDef().isSubClassOf("SideEffectsTraitBase");64}65