1 //===- Type.cpp - Type class ----------------------------------------------===//
2 //
3 // Copyright 2019 The MLIR Authors.
4 //
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
8 //
9 //   http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16 // =============================================================================
17 //
18 // Type wrapper to simplify using TableGen Record defining a MLIR Type.
19 //
20 //===----------------------------------------------------------------------===//
21 
22 #include "mlir/TableGen/Type.h"
23 #include "llvm/TableGen/Record.h"
24 
25 using namespace mlir;
26 using namespace mlir::tblgen;
27 
28 TypeConstraint::TypeConstraint(const llvm::Record *record)
29     : Constraint(Constraint::CK_Type, record) {
30   assert(def->isSubClassOf("TypeConstraint") &&
31          "must be subclass of TableGen 'TypeConstraint' class");
32 }
33 
34 TypeConstraint::TypeConstraint(const llvm::DefInit *init)
35     : TypeConstraint(init->getDef()) {}
36 
37 bool TypeConstraint::isVariadic() const {
38   return def->isSubClassOf("Variadic");
39 }
40 
41 Type::Type(const llvm::Record *record) : TypeConstraint(record) {}
42 
43 StringRef Type::getTypeDescription() const {
44   return def->getValueAsString("typeDescription");
45 }
46 
47 Dialect Type::getDialect() const {
48   return Dialect(def->getValueAsDef("dialect"));
49 }
50