Files
llvm/mlir/lib/TableGen/Property.cpp
Mehdi Amini 660f714e26 [MLIR] Add native Bytecode support for properties
This is adding a new interface (`BytecodeOpInterface`) to allow operations to
opt-in skipping conversion to attribute and serializing properties to native
bytecode.

The scheme relies on a new section where properties are stored in sequence

  { size, serialize_properties }, ...

The operations are storing the index of a properties, a table of offset is
built when loading the properties section the first time.

This is a re-commit of 837d1ce0dc which conflicted with another patch upgrading
the bytecode and the collision wasn't properly resolved before.

Differential Revision: https://reviews.llvm.org/D151065
2023-05-26 17:45:01 -07:00

97 lines
3.0 KiB
C++

//===- Property.cpp - Property wrapper 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
//
//===----------------------------------------------------------------------===//
//
// Property wrapper to simplify using TableGen Record defining a MLIR
// Property.
//
//===----------------------------------------------------------------------===//
#include "mlir/TableGen/Property.h"
#include "mlir/TableGen/Format.h"
#include "mlir/TableGen/Operator.h"
#include "llvm/TableGen/Record.h"
using namespace mlir;
using namespace mlir::tblgen;
using llvm::DefInit;
using llvm::Init;
using llvm::Record;
using llvm::StringInit;
// Returns the initializer's value as string if the given TableGen initializer
// is a code or string initializer. Returns the empty StringRef otherwise.
static StringRef getValueAsString(const Init *init) {
if (const auto *str = dyn_cast<StringInit>(init))
return str->getValue().trim();
return {};
}
Property::Property(const Record *record) : def(record) {
assert((record->isSubClassOf("Property") || record->isSubClassOf("Attr")) &&
"must be subclass of TableGen 'Property' class");
}
Property::Property(const DefInit *init) : Property(init->getDef()) {}
StringRef Property::getStorageType() const {
const auto *init = def->getValueInit("storageType");
auto type = getValueAsString(init);
if (type.empty())
return "Property";
return type;
}
StringRef Property::getInterfaceType() const {
const auto *init = def->getValueInit("interfaceType");
return getValueAsString(init);
}
StringRef Property::getConvertFromStorageCall() const {
const auto *init = def->getValueInit("convertFromStorage");
return getValueAsString(init);
}
StringRef Property::getAssignToStorageCall() const {
const auto *init = def->getValueInit("assignToStorage");
return getValueAsString(init);
}
StringRef Property::getConvertToAttributeCall() const {
const auto *init = def->getValueInit("convertToAttribute");
return getValueAsString(init);
}
StringRef Property::getConvertFromAttributeCall() const {
const auto *init = def->getValueInit("convertFromAttribute");
return getValueAsString(init);
}
StringRef Property::getReadFromMlirBytecodeCall() const {
const auto *init = def->getValueInit("readFromMlirBytecode");
return getValueAsString(init);
}
StringRef Property::getWriteToMlirBytecodeCall() const {
const auto *init = def->getValueInit("writeToMlirBytecode");
return getValueAsString(init);
}
StringRef Property::getHashPropertyCall() const {
return getValueAsString(def->getValueInit("hashProperty"));
}
bool Property::hasDefaultValue() const { return !getDefaultValue().empty(); }
StringRef Property::getDefaultValue() const {
const auto *init = def->getValueInit("defaultValue");
return getValueAsString(init);
}
const llvm::Record &Property::getDef() const { return *def; }