mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-25 13:46:07 +00:00
b0774e5f50
Extra definitions are placed in the generated source file for each op class. The substitution `$cppClass` is replaced by the op's C++ class name.
This is useful when declaring but not defining methods in TableGen base classes:
```
class BaseOp<string mnemonic>
: Op<MyDialect, mnemonic, [DeclareOpInterfaceMethods<SomeInterface>] {
let extraClassDeclaration = [{
// ZOp is declared at at the bottom of the file and is incomplete here
ZOp getParent();
}];
let extraClassDefinition = [{
int $cppClass::someInterfaceMethod() {
return someUtilityFunction(*this);
}
ZOp $cppClass::getParent() {
return dyn_cast<ZOp>(this->getParentOp());
}
}];
}
```
Certain things may prevent defining these functions inline, in the declaration. In this example, `ZOp` in the same dialect is incomplete at the function declaration because ops classes are declared in alphabetical order. Alternatively, functions may be too big to be desired as inlined, or they may require dependencies that create cyclic includes, or they may be calling a templated utility function that one may not want to expose in a header. If the functions are not inlined, then inheriting from the base class N times means that each function will need to be defined N times. With `extraClassDefinitions`, they only need to be defined once.
Reviewed By: rriddle
Differential Revision: https://reviews.llvm.org/D115783
37 lines
1.4 KiB
C++
37 lines
1.4 KiB
C++
//===- OpClass.cpp - 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "OpClass.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::tblgen;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// OpClass definitions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
OpClass::OpClass(StringRef name, StringRef extraClassDeclaration,
|
|
std::string extraClassDefinition)
|
|
: Class(name.str()), extraClassDeclaration(extraClassDeclaration),
|
|
extraClassDefinition(std::move(extraClassDefinition)),
|
|
parent(addParent("::mlir::Op")) {
|
|
parent.addTemplateParam(getClassName().str());
|
|
declare<VisibilityDeclaration>(Visibility::Public);
|
|
/// Inherit functions from Op.
|
|
declare<UsingDeclaration>("Op::Op");
|
|
declare<UsingDeclaration>("Op::print");
|
|
/// Type alias for the adaptor class.
|
|
declare<UsingDeclaration>("Adaptor", className + "Adaptor");
|
|
}
|
|
|
|
void OpClass::finalize() {
|
|
Class::finalize();
|
|
declare<VisibilityDeclaration>(Visibility::Public);
|
|
declare<ExtraClassDeclaration>(extraClassDeclaration, extraClassDefinition);
|
|
}
|