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
|
//===- OpClass.h - Implementation of an Op Class --------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#ifndef MLIR_TOOLS_MLIRTBLGEN_OPCLASS_H_
#define MLIR_TOOLS_MLIRTBLGEN_OPCLASS_H_
#include "mlir/TableGen/Class.h"
namespace mlir {
namespace tblgen {
/// Class for holding an op for C++ code emission. The class is specialized to
/// add Op-specific declarations to the class.
class OpClass : public Class {
public:
/// Create an operation class with extra class declarations, whose default
/// visibility is public. Also declares at the top of the class:
///
/// - inheritance of constructors from `Op`
/// - inheritance of `print`
/// - a type alias for the associated adaptor class
///
OpClass(StringRef name, std::string extraClassDeclaration,
std::string extraClassDefinition);
/// Add an op trait.
void addTrait(Twine trait) { parent.addTemplateParam(trait.str()); }
/// The operation class is finalized by calling `Class::finalize` to delcare
/// all pending private and public methods (ops don't have custom constructors
/// or fields). Then, the extra class declarations are appended to the end of
/// the class declaration.
void finalize() override;
private:
/// Hand-written extra class declarations.
std::string extraClassDeclaration;
/// Hand-written extra class definitions.
std::string extraClassDefinition;
/// The parent class, which also contains the traits to be inherited.
ParentClass &parent;
};
} // namespace tblgen
} // namespace mlir
#endif // MLIR_TOOLS_MLIRTBLGEN_OPCLASS_H_
|
