104 lines
3.8 KiB
C++
104 lines
3.8 KiB
C++
/*===- TableGen'erated file -------------------------------------*- C++ -*-===*\
|
|
|* *|
|
|
|* Enum Utility Declarations *|
|
|
|* *|
|
|
|* Automatically generated file, do not edit! *|
|
|
|* From: StructuredOpsUtils.td *|
|
|
|* *|
|
|
\*===----------------------------------------------------------------------===*/
|
|
|
|
namespace mlir::utils {
|
|
|
|
// Iterator type
|
|
enum class IteratorType : uint32_t {
|
|
parallel = 0,
|
|
reduction = 1,
|
|
};
|
|
|
|
::std::optional<IteratorType> symbolizeIteratorType(uint32_t);
|
|
::llvm::StringRef stringifyIteratorType(IteratorType);
|
|
::std::optional<IteratorType> symbolizeIteratorType(::llvm::StringRef);
|
|
inline constexpr unsigned getMaxEnumValForIteratorType() {
|
|
return 1;
|
|
}
|
|
|
|
|
|
inline ::llvm::StringRef stringifyEnum(IteratorType enumValue) {
|
|
return stringifyIteratorType(enumValue);
|
|
}
|
|
|
|
template <typename EnumType>
|
|
::std::optional<EnumType> symbolizeEnum(::llvm::StringRef);
|
|
|
|
template <>
|
|
inline ::std::optional<IteratorType> symbolizeEnum<IteratorType>(::llvm::StringRef str) {
|
|
return symbolizeIteratorType(str);
|
|
}
|
|
|
|
} // namespace mlir::utils
|
|
|
|
namespace mlir {
|
|
template <typename T, typename>
|
|
struct FieldParser;
|
|
|
|
template<>
|
|
struct FieldParser<::mlir::utils::IteratorType, ::mlir::utils::IteratorType> {
|
|
template <typename ParserT>
|
|
static FailureOr<::mlir::utils::IteratorType> parse(ParserT &parser) {
|
|
// Parse the keyword/string containing the enum.
|
|
std::string enumKeyword;
|
|
auto loc = parser.getCurrentLocation();
|
|
if (failed(parser.parseOptionalKeywordOrString(&enumKeyword)))
|
|
return parser.emitError(loc, "expected keyword for Iterator type");
|
|
|
|
// Symbolize the keyword.
|
|
if (::std::optional<::mlir::utils::IteratorType> attr = ::mlir::utils::symbolizeEnum<::mlir::utils::IteratorType>(enumKeyword))
|
|
return *attr;
|
|
return parser.emitError(loc, "expected one of [parallel, reduction] for Iterator type, got: ") << enumKeyword;
|
|
}
|
|
};
|
|
|
|
/// Support for std::optional, useful in attribute/type definition where the enum is
|
|
/// used as:
|
|
///
|
|
/// let parameters = (ins OptionalParameter<"std::optional<TheEnumName>">:$value);
|
|
template<>
|
|
struct FieldParser<std::optional<::mlir::utils::IteratorType>, std::optional<::mlir::utils::IteratorType>> {
|
|
template <typename ParserT>
|
|
static FailureOr<std::optional<::mlir::utils::IteratorType>> parse(ParserT &parser) {
|
|
// Parse the keyword/string containing the enum.
|
|
std::string enumKeyword;
|
|
auto loc = parser.getCurrentLocation();
|
|
if (failed(parser.parseOptionalKeywordOrString(&enumKeyword, {"parallel", "reduction"})))
|
|
return std::optional<::mlir::utils::IteratorType>{};
|
|
|
|
// Symbolize the keyword.
|
|
if (::std::optional<::mlir::utils::IteratorType> attr = ::mlir::utils::symbolizeEnum<::mlir::utils::IteratorType>(enumKeyword))
|
|
return attr;
|
|
return parser.emitError(loc, "expected one of [parallel, reduction] for Iterator type, got: ") << enumKeyword;
|
|
}
|
|
};
|
|
} // namespace mlir
|
|
|
|
namespace llvm {
|
|
inline ::llvm::raw_ostream &operator<<(::llvm::raw_ostream &p, ::mlir::utils::IteratorType value) {
|
|
auto valueStr = stringifyEnum(value);
|
|
return p << valueStr;
|
|
}
|
|
} // namespace llvm
|
|
|
|
namespace llvm {
|
|
template<> struct DenseMapInfo<::mlir::utils::IteratorType> {
|
|
using StorageInfo = ::llvm::DenseMapInfo<uint32_t>;
|
|
|
|
static unsigned getHashValue(const ::mlir::utils::IteratorType &val) {
|
|
return StorageInfo::getHashValue(static_cast<uint32_t>(val));
|
|
}
|
|
|
|
static bool isEqual(const ::mlir::utils::IteratorType &lhs, const ::mlir::utils::IteratorType &rhs) {
|
|
return lhs == rhs;
|
|
}
|
|
};
|
|
}
|
|
|