mlir/lib/Dialect/IRDL/IRDLSymbols.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

//===- IRDLSymbols.cpp - IRDL-related symbol logic --------------*- C++ -*-===//
//
// This file is licensed 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
//
//===----------------------------------------------------------------------===//

#include "mlir/Dialect/IRDL/IRDLSymbols.h"
#include "mlir/Dialect/IRDL/IR/IRDL.h"

using namespace mlir;
using namespace mlir::irdl;

static Operation *lookupDialectOp(Operation *source) {
  Operation *dialectOp = source;
  while (dialectOp && !isa<DialectOp>(dialectOp))
    dialectOp = dialectOp->getParentOp();

  if (!dialectOp)
    llvm_unreachable("symbol lookup near dialect must originate from "
                     "within a dialect definition");

  return dialectOp;
}

Operation *
mlir::irdl::lookupSymbolNearDialect(SymbolTableCollection &symbolTable,
                                    Operation *source, SymbolRefAttr symbol) {
  return symbolTable.lookupNearestSymbolFrom(
      lookupDialectOp(source)->getParentOp(), symbol);
}

Operation *mlir::irdl::lookupSymbolNearDialect(Operation *source,
                                               SymbolRefAttr symbol) {
  return SymbolTable::lookupNearestSymbolFrom(
      lookupDialectOp(source)->getParentOp(), symbol);
}