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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
//===- Trait.cpp ----------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// Trait wrapper to simplify using TableGen Record defining a MLIR Trait.
//
//===----------------------------------------------------------------------===//
#include "mlir/TableGen/Trait.h"
#include "mlir/TableGen/Interfaces.h"
#include "mlir/TableGen/Predicate.h"
#include "llvm/ADT/StringExtras.h"
#include "llvm/Support/FormatVariadic.h"
#include "llvm/TableGen/Error.h"
#include "llvm/TableGen/Record.h"
using namespace mlir;
using namespace mlir::tblgen;
//===----------------------------------------------------------------------===//
// Trait
//===----------------------------------------------------------------------===//
Trait Trait::create(const llvm::Init *init) {
auto *def = cast<llvm::DefInit>(init)->getDef();
if (def->isSubClassOf("PredTrait"))
return Trait(Kind::Pred, def);
if (def->isSubClassOf("GenInternalTrait"))
return Trait(Kind::Internal, def);
if (def->isSubClassOf("InterfaceTrait"))
return Trait(Kind::Interface, def);
assert(def->isSubClassOf("NativeTrait"));
return Trait(Kind::Native, def);
}
Trait::Trait(Kind kind, const llvm::Record *def) : def(def), kind(kind) {}
//===----------------------------------------------------------------------===//
// NativeTrait
//===----------------------------------------------------------------------===//
std::string NativeTrait::getFullyQualifiedTraitName() const {
llvm::StringRef trait = def->getValueAsString("trait");
llvm::StringRef cppNamespace = def->getValueAsString("cppNamespace");
return cppNamespace.empty() ? trait.str()
: (cppNamespace + "::" + trait).str();
}
bool NativeTrait::isStructuralOpTrait() const {
return def->isSubClassOf("StructuralOpTrait");
}
StringRef NativeTrait::getExtraConcreteClassDeclaration() const {
return def->getValueAsString("extraConcreteClassDeclaration");
}
StringRef NativeTrait::getExtraConcreteClassDefinition() const {
return def->getValueAsString("extraConcreteClassDefinition");
}
//===----------------------------------------------------------------------===//
// InternalTrait
//===----------------------------------------------------------------------===//
llvm::StringRef InternalTrait::getFullyQualifiedTraitName() const {
return def->getValueAsString("trait");
}
//===----------------------------------------------------------------------===//
// PredTrait
//===----------------------------------------------------------------------===//
std::string PredTrait::getPredTemplate() const {
auto pred = Pred(def->getValueInit("predicate"));
return pred.getCondition();
}
llvm::StringRef PredTrait::getSummary() const {
return def->getValueAsString("summary");
}
//===----------------------------------------------------------------------===//
// InterfaceTrait
//===----------------------------------------------------------------------===//
Interface InterfaceTrait::getInterface() const { return Interface(def); }
std::string InterfaceTrait::getFullyQualifiedTraitName() const {
llvm::StringRef trait = def->getValueAsString("trait");
llvm::StringRef cppNamespace = def->getValueAsString("cppNamespace");
return cppNamespace.empty() ? trait.str()
: (cppNamespace + "::" + trait).str();
}
bool InterfaceTrait::shouldDeclareMethods() const {
return def->isSubClassOf("DeclareInterfaceMethods");
}
std::vector<StringRef> InterfaceTrait::getAlwaysDeclaredMethods() const {
return def->getValueAsListOfStrings("alwaysOverriddenMethods");
}
|
