xref: /llvm-project/mlir/lib/TableGen/Trait.cpp (revision 47b0a9b9311ffdaab511c240de89b5da75b1252b) 
1 //===- Trait.cpp ----------------------------------------------------------===//
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 // Trait wrapper to simplify using TableGen Record defining a MLIR Trait.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "mlir/TableGen/Trait.h"
14 #include "mlir/TableGen/Interfaces.h"
15 #include "mlir/TableGen/Predicate.h"
16 #include "llvm/ADT/StringExtras.h"
17 #include "llvm/Support/FormatVariadic.h"
18 #include "llvm/TableGen/Error.h"
19 #include "llvm/TableGen/Record.h"
20 
21 using namespace mlir;
22 using namespace mlir::tblgen;
23 
24 //===----------------------------------------------------------------------===//
25 // Trait
26 //===----------------------------------------------------------------------===//
27 
create(const llvm::Init * init)28 Trait Trait::create(const llvm::Init *init) {
29   auto *def = cast<llvm::DefInit>(init)->getDef();
30   if (def->isSubClassOf("PredTrait"))
31     return Trait(Kind::Pred, def);
32   if (def->isSubClassOf("GenInternalTrait"))
33     return Trait(Kind::Internal, def);
34   if (def->isSubClassOf("InterfaceTrait"))
35     return Trait(Kind::Interface, def);
36   assert(def->isSubClassOf("NativeTrait"));
37   return Trait(Kind::Native, def);
38 }
39 
Trait(Kind kind,const llvm::Record * def)40 Trait::Trait(Kind kind, const llvm::Record *def) : def(def), kind(kind) {}
41 
42 //===----------------------------------------------------------------------===//
43 // NativeTrait
44 //===----------------------------------------------------------------------===//
45 
getFullyQualifiedTraitName() const46 std::string NativeTrait::getFullyQualifiedTraitName() const {
47   llvm::StringRef trait = def->getValueAsString("trait");
48   llvm::StringRef cppNamespace = def->getValueAsString("cppNamespace");
49   return cppNamespace.empty() ? trait.str()
50                               : (cppNamespace + "::" + trait).str();
51 }
52 
isStructuralOpTrait() const53 bool NativeTrait::isStructuralOpTrait() const {
54   return def->isSubClassOf("StructuralOpTrait");
55 }
56 
getExtraConcreteClassDeclaration() const57 StringRef NativeTrait::getExtraConcreteClassDeclaration() const {
58   return def->getValueAsString("extraConcreteClassDeclaration");
59 }
60 
getExtraConcreteClassDefinition() const61 StringRef NativeTrait::getExtraConcreteClassDefinition() const {
62   return def->getValueAsString("extraConcreteClassDefinition");
63 }
64 
65 //===----------------------------------------------------------------------===//
66 // InternalTrait
67 //===----------------------------------------------------------------------===//
68 
getFullyQualifiedTraitName() const69 llvm::StringRef InternalTrait::getFullyQualifiedTraitName() const {
70   return def->getValueAsString("trait");
71 }
72 
73 //===----------------------------------------------------------------------===//
74 // PredTrait
75 //===----------------------------------------------------------------------===//
76 
getPredTemplate() const77 std::string PredTrait::getPredTemplate() const {
78   auto pred = Pred(def->getValueInit("predicate"));
79   return pred.getCondition();
80 }
81 
getSummary() const82 llvm::StringRef PredTrait::getSummary() const {
83   return def->getValueAsString("summary");
84 }
85 
86 //===----------------------------------------------------------------------===//
87 // InterfaceTrait
88 //===----------------------------------------------------------------------===//
89 
getInterface() const90 Interface InterfaceTrait::getInterface() const { return Interface(def); }
91 
getFullyQualifiedTraitName() const92 std::string InterfaceTrait::getFullyQualifiedTraitName() const {
93   llvm::StringRef trait = def->getValueAsString("trait");
94   llvm::StringRef cppNamespace = def->getValueAsString("cppNamespace");
95   return cppNamespace.empty() ? trait.str()
96                               : (cppNamespace + "::" + trait).str();
97 }
98 
shouldDeclareMethods() const99 bool InterfaceTrait::shouldDeclareMethods() const {
100   return def->isSubClassOf("DeclareInterfaceMethods");
101 }
102 
getAlwaysDeclaredMethods() const103 std::vector<StringRef> InterfaceTrait::getAlwaysDeclaredMethods() const {
104   return def->getValueAsListOfStrings("alwaysOverriddenMethods");
105 }
106