Skip to content

[NFC][MLIR][TableGen] Adopt NamespaceEmitter more widely #162015

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
+17 −77
Open

[NFC][MLIR][TableGen] Adopt NamespaceEmitter more widely #162015

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 4 additions & 17 deletions mlir/tools/mlir-tblgen/EnumsGen.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
#include "llvm/ADT/StringExtras.h"
#include "llvm/Support/FormatVariadic.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/TableGen/CodeGenHelpers.h"
#include "llvm/TableGen/Error.h"
#include "llvm/TableGen/Record.h"
#include "llvm/TableGen/TableGenBackend.h"
Expand Down Expand Up @@ -703,11 +704,7 @@ static void emitEnumDecl(const Record &enumDef, raw_ostream &os) {
StringRef underlyingToSymFnName = enumInfo.getUnderlyingToSymbolFnName();
auto enumerants = enumInfo.getAllCases();

SmallVector<StringRef, 2> namespaces;
llvm::SplitString(cppNamespace, namespaces, "::");

for (auto ns : namespaces)
os << "namespace " << ns << " {\n";
llvm::NamespaceEmitter ns(os, cppNamespace);

// Emit the enum class definition
emitEnumClass(enumDef, enumName, underlyingType, description, enumerants, os);
Expand Down Expand Up @@ -768,8 +765,7 @@ class {1} : public ::mlir::{2} {
os << formatv(attrClassDecl, enumName, attrClassName, baseAttrClassName);
}

for (auto ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
ns.close();

// Generate a generic parser and printer for the enum.
std::string qualName =
Expand All @@ -792,13 +788,8 @@ static bool emitEnumDecls(const RecordKeeper &records, raw_ostream &os) {

static void emitEnumDef(const Record &enumDef, raw_ostream &os) {
EnumInfo enumInfo(enumDef);
StringRef cppNamespace = enumInfo.getCppNamespace();

SmallVector<StringRef, 2> namespaces;
llvm::SplitString(cppNamespace, namespaces, "::");

for (auto ns : namespaces)
os << "namespace " << ns << " {\n";
llvm::NamespaceEmitter ns(os, enumInfo.getCppNamespace());

if (enumInfo.isBitEnum()) {
emitSymToStrFnForBitEnum(enumDef, os);
Expand All @@ -812,10 +803,6 @@ static void emitEnumDef(const Record &enumDef, raw_ostream &os) {

if (enumInfo.genSpecializedAttr())
emitSpecializedAttrDef(enumDef, os);

for (auto ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
os << "\n";
}

static bool emitEnumDefs(const RecordKeeper &records, raw_ostream &os) {
Expand Down
38 changes: 6 additions & 32 deletions mlir/tools/mlir-tblgen/OpInterfacesGen.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
#include "llvm/ADT/StringExtras.h"
#include "llvm/Support/FormatVariadic.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/TableGen/CodeGenHelpers.h"
#include "llvm/TableGen/Error.h"
#include "llvm/TableGen/Record.h"
#include "llvm/TableGen/TableGenBackend.h"
Expand Down Expand Up @@ -340,11 +341,7 @@ void InterfaceGenerator::emitModelDecl(const Interface &interface) {
}

void InterfaceGenerator::emitModelMethodsDef(const Interface &interface) {
llvm::SmallVector<StringRef, 2> namespaces;
llvm::SplitString(interface.getCppNamespace(), namespaces, "::");
for (StringRef ns : namespaces)
os << "namespace " << ns << " {\n";

llvm::NamespaceEmitter ns(os, interface.getCppNamespace());
for (auto &method : interface.getMethods()) {
os << "template<typename " << valueTemplate << ">\n";
emitCPPType(method.getReturnType(), os);
Expand Down Expand Up @@ -440,18 +437,11 @@ void InterfaceGenerator::emitModelMethodsDef(const Interface &interface) {
method.isStatic() ? &ctx : &nonStaticMethodFmt);
os << "\n}\n";
}

for (StringRef ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
}

void InterfaceGenerator::emitInterfaceTraitDecl(const Interface &interface) {
llvm::SmallVector<StringRef, 2> namespaces;
llvm::SplitString(interface.getCppNamespace(), namespaces, "::");
for (StringRef ns : namespaces)
os << "namespace " << ns << " {\n";

os << "namespace detail {\n";
auto cppNamespace = (interface.getCppNamespace() + "::detail").str();
llvm::NamespaceEmitter ns(os, cppNamespace);

StringRef interfaceName = interface.getName();
auto interfaceTraitsName = (interfaceName + "InterfaceTraits").str();
Expand Down Expand Up @@ -501,10 +491,6 @@ void InterfaceGenerator::emitInterfaceTraitDecl(const Interface &interface) {
os << tblgen::tgfmt(*extraTraitDecls, &traitMethodFmt) << "\n";

os << " };\n";
os << "}// namespace detail\n";

for (StringRef ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
}

static void emitInterfaceDeclMethods(const Interface &interface,
Expand All @@ -529,10 +515,7 @@ static void emitInterfaceDeclMethods(const Interface &interface,
}

void InterfaceGenerator::forwardDeclareInterface(const Interface &interface) {
llvm::SmallVector<StringRef, 2> namespaces;
llvm::SplitString(interface.getCppNamespace(), namespaces, "::");
for (StringRef ns : namespaces)
os << "namespace " << ns << " {\n";
llvm::NamespaceEmitter ns(os, interface.getCppNamespace());

// Emit a forward declaration of the interface class so that it becomes usable
// in the signature of its methods.
Expand All @@ -544,16 +527,10 @@ void InterfaceGenerator::forwardDeclareInterface(const Interface &interface) {

StringRef interfaceName = interface.getName();
os << "class " << interfaceName << ";\n";

for (StringRef ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
}

void InterfaceGenerator::emitInterfaceDecl(const Interface &interface) {
llvm::SmallVector<StringRef, 2> namespaces;
llvm::SplitString(interface.getCppNamespace(), namespaces, "::");
for (StringRef ns : namespaces)
os << "namespace " << ns << " {\n";
llvm::NamespaceEmitter ns(os, interface.getCppNamespace());

StringRef interfaceName = interface.getName();
auto interfaceTraitsName = (interfaceName + "InterfaceTraits").str();
Expand Down Expand Up @@ -633,9 +610,6 @@ void InterfaceGenerator::emitInterfaceDecl(const Interface &interface) {
}

os << "};\n";

for (StringRef ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
}

bool InterfaceGenerator::emitInterfaceDecls() {
Expand Down
35 changes: 7 additions & 28 deletions mlir/tools/mlir-tblgen/SPIRVUtilsGen.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -259,8 +259,8 @@ static void emitInterfaceDecl(const Availability &availability,
std::string interfaceTraitsName =
std::string(formatv("{0}Traits", interfaceName));

StringRef cppNamespace = availability.getInterfaceClassNamespace();
llvm::NamespaceEmitter nsEmitter(os, cppNamespace);
llvm::NamespaceEmitter nsEmitter(os,
availability.getInterfaceClassNamespace());
os << "class " << interfaceName << ";\n\n";

// Emit the traits struct containing the concept and model declarations.
Expand Down Expand Up @@ -418,15 +418,9 @@ static void emitAvailabilityQueryForBitEnum(const Record &enumDef,
static void emitEnumDecl(const Record &enumDef, raw_ostream &os) {
EnumInfo enumInfo(enumDef);
StringRef enumName = enumInfo.getEnumClassName();
StringRef cppNamespace = enumInfo.getCppNamespace();
auto enumerants = enumInfo.getAllCases();

llvm::SmallVector<StringRef, 2> namespaces;
llvm::SplitString(cppNamespace, namespaces, "::");

for (auto ns : namespaces)
os << "namespace " << ns << " {\n";

llvm::NamespaceEmitter ns(os, enumInfo.getCppNamespace());
llvm::StringSet<> handledClasses;

// Place all availability specifications to their corresponding
Expand All @@ -441,9 +435,6 @@ static void emitEnumDecl(const Record &enumDef, raw_ostream &os) {
enumName);
handledClasses.insert(className);
}

for (auto ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
}

static bool emitEnumDecls(const RecordKeeper &records, raw_ostream &os) {
Expand All @@ -459,31 +450,19 @@ static bool emitEnumDecls(const RecordKeeper &records, raw_ostream &os) {

static void emitEnumDef(const Record &enumDef, raw_ostream &os) {
EnumInfo enumInfo(enumDef);
StringRef cppNamespace = enumInfo.getCppNamespace();

llvm::SmallVector<StringRef, 2> namespaces;
llvm::SplitString(cppNamespace, namespaces, "::");

for (auto ns : namespaces)
os << "namespace " << ns << " {\n";
llvm::NamespaceEmitter ns(os, enumInfo.getCppNamespace());

if (enumInfo.isBitEnum()) {
if (enumInfo.isBitEnum())
emitAvailabilityQueryForBitEnum(enumDef, os);
} else {
else
emitAvailabilityQueryForIntEnum(enumDef, os);
}

for (auto ns : llvm::reverse(namespaces))
os << "} // namespace " << ns << "\n";
os << "\n";
}

static bool emitEnumDefs(const RecordKeeper &records, raw_ostream &os) {
llvm::emitSourceFileHeader("SPIR-V Enum Availability Definitions", os,
records);

auto defs = records.getAllDerivedDefinitions("EnumInfo");
for (const auto *def : defs)
for (const Record *def : records.getAllDerivedDefinitions("EnumInfo"))
emitEnumDef(*def, os);

return false;
Expand Down