mlir/lib/Conversion/PDLToPDLInterp/Predicate.cpp

   1 //===- Predicate.cpp - Pattern predicates ---------------------------------===//
   2 //
   3 // Part of the LLVM 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-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "Predicate.h"
  10
  11 using namespace mlir;
  12 using namespace mlir::pdl_to_pdl_interp;
  13
  14 //===----------------------------------------------------------------------===//
  15 // Positions
  16 //===----------------------------------------------------------------------===//
  17
  18 Position::~Position() = default;
  19
  20 /// Returns the depth of the first ancestor operation position.
  21 unsigned Position::getOperationDepth() const {
  22   if (const auto *operationPos = dyn_cast<OperationPosition>(this))
  23     return operationPos->getDepth();
  24   return parent ? parent->getOperationDepth() : 0;
  25 }
  26
  27 //===----------------------------------------------------------------------===//
  28 // AttributePosition
  29
  30 AttributePosition::AttributePosition(const KeyTy &key) : Base(key) {
  31   parent = key.first;
  32 }
  33
  34 //===----------------------------------------------------------------------===//
  35 // OperandPosition
  36
  37 OperandPosition::OperandPosition(const KeyTy &key) : Base(key) {
  38   parent = key.first;
  39 }
  40
  41 //===----------------------------------------------------------------------===//
  42 // OperandGroupPosition
  43
  44 OperandGroupPosition::OperandGroupPosition(const KeyTy &key) : Base(key) {
  45   parent = std::get<0>(key);
  46 }
  47
  48 //===----------------------------------------------------------------------===//
  49 // OperationPosition
  50
  51 bool OperationPosition::isOperandDefiningOp() const {
  52   return isa_and_nonnull<OperandPosition, OperandGroupPosition>(parent);
  53 }