e95e94adc6
This PR massively reorganizes the Test dialect's source files. It moves manually-written op hooks into `TestOpDefs.cpp`, moves format custom directive parsers and printers into `TestFormatUtils`, adds missing comment blocks, and moves around where generated source files are included for types, attributes, enums, etc. into their own source file. This will hopefully help navigate the test dialect source code, but also speeds up compile time of the test dialect by putting generated source files into separate compilation units. This also sets up the test dialect to shard its op definitions, done in the next PR.
212 lines
9.5 KiB
C++
212 lines
9.5 KiB
C++
//===- TestFormatUtils.h - MLIR Test Dialect Assembly Format Utilities ----===//
|
|
//
|
|
// 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_TESTFORMATUTILS_H
|
|
#define MLIR_TESTFORMATUTILS_H
|
|
|
|
#include "mlir/IR/OpImplementation.h"
|
|
|
|
namespace test {
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveOperands
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveOperands(
|
|
mlir::OpAsmParser &parser, mlir::OpAsmParser::UnresolvedOperand &operand,
|
|
std::optional<mlir::OpAsmParser::UnresolvedOperand> &optOperand,
|
|
llvm::SmallVectorImpl<mlir::OpAsmParser::UnresolvedOperand> &varOperands);
|
|
|
|
void printCustomDirectiveOperands(mlir::OpAsmPrinter &printer,
|
|
mlir::Operation *, mlir::Value operand,
|
|
mlir::Value optOperand,
|
|
mlir::OperandRange varOperands);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveResults
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult
|
|
parseCustomDirectiveResults(mlir::OpAsmParser &parser, mlir::Type &operandType,
|
|
mlir::Type &optOperandType,
|
|
llvm::SmallVectorImpl<mlir::Type> &varOperandTypes);
|
|
|
|
void printCustomDirectiveResults(mlir::OpAsmPrinter &printer, mlir::Operation *,
|
|
mlir::Type operandType,
|
|
mlir::Type optOperandType,
|
|
mlir::TypeRange varOperandTypes);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveWithTypeRefs
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveWithTypeRefs(
|
|
mlir::OpAsmParser &parser, mlir::Type operandType,
|
|
mlir::Type optOperandType,
|
|
const llvm::SmallVectorImpl<mlir::Type> &varOperandTypes);
|
|
|
|
void printCustomDirectiveWithTypeRefs(mlir::OpAsmPrinter &printer,
|
|
mlir::Operation *op,
|
|
mlir::Type operandType,
|
|
mlir::Type optOperandType,
|
|
mlir::TypeRange varOperandTypes);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveOperandsAndTypes
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveOperandsAndTypes(
|
|
mlir::OpAsmParser &parser, mlir::OpAsmParser::UnresolvedOperand &operand,
|
|
std::optional<mlir::OpAsmParser::UnresolvedOperand> &optOperand,
|
|
llvm::SmallVectorImpl<mlir::OpAsmParser::UnresolvedOperand> &varOperands,
|
|
mlir::Type &operandType, mlir::Type &optOperandType,
|
|
llvm::SmallVectorImpl<mlir::Type> &varOperandTypes);
|
|
|
|
void printCustomDirectiveOperandsAndTypes(
|
|
mlir::OpAsmPrinter &printer, mlir::Operation *op, mlir::Value operand,
|
|
mlir::Value optOperand, mlir::OperandRange varOperands,
|
|
mlir::Type operandType, mlir::Type optOperandType,
|
|
mlir::TypeRange varOperandTypes);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveRegions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveRegions(
|
|
mlir::OpAsmParser &parser, mlir::Region ®ion,
|
|
llvm::SmallVectorImpl<std::unique_ptr<mlir::Region>> &varRegions);
|
|
|
|
void printCustomDirectiveRegions(
|
|
mlir::OpAsmPrinter &printer, mlir::Operation *, mlir::Region ®ion,
|
|
llvm::MutableArrayRef<mlir::Region> varRegions);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveSuccessors
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveSuccessors(
|
|
mlir::OpAsmParser &parser, mlir::Block *&successor,
|
|
llvm::SmallVectorImpl<mlir::Block *> &varSuccessors);
|
|
|
|
void printCustomDirectiveSuccessors(mlir::OpAsmPrinter &printer,
|
|
mlir::Operation *, mlir::Block *successor,
|
|
mlir::SuccessorRange varSuccessors);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveAttributes
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveAttributes(mlir::OpAsmParser &parser,
|
|
mlir::IntegerAttr &attr,
|
|
mlir::IntegerAttr &optAttr);
|
|
|
|
void printCustomDirectiveAttributes(mlir::OpAsmPrinter &printer,
|
|
mlir::Operation *,
|
|
mlir::Attribute attribute,
|
|
mlir::Attribute optAttribute);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveAttrDict
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveAttrDict(mlir::OpAsmParser &parser,
|
|
mlir::NamedAttrList &attrs);
|
|
|
|
void printCustomDirectiveAttrDict(mlir::OpAsmPrinter &printer,
|
|
mlir::Operation *op,
|
|
mlir::DictionaryAttr attrs);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveOptionalOperandRef
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomDirectiveOptionalOperandRef(
|
|
mlir::OpAsmParser &parser,
|
|
std::optional<mlir::OpAsmParser::UnresolvedOperand> &optOperand);
|
|
|
|
void printCustomDirectiveOptionalOperandRef(mlir::OpAsmPrinter &printer,
|
|
mlir::Operation *op,
|
|
mlir::Value optOperand);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveOptionalOperand
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseCustomOptionalOperand(
|
|
mlir::OpAsmParser &parser,
|
|
std::optional<mlir::OpAsmParser::UnresolvedOperand> &optOperand);
|
|
|
|
void printCustomOptionalOperand(mlir::OpAsmPrinter &printer, mlir::Operation *,
|
|
mlir::Value optOperand);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveSwitchCases
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseSwitchCases(
|
|
mlir::OpAsmParser &p, mlir::DenseI64ArrayAttr &cases,
|
|
llvm::SmallVectorImpl<std::unique_ptr<mlir::Region>> &caseRegions);
|
|
|
|
void printSwitchCases(mlir::OpAsmPrinter &p, mlir::Operation *op,
|
|
mlir::DenseI64ArrayAttr cases,
|
|
mlir::RegionRange caseRegions);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomUsingPropertyInCustom
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
bool parseUsingPropertyInCustom(mlir::OpAsmParser &parser, int64_t value[3]);
|
|
|
|
void printUsingPropertyInCustom(mlir::OpAsmPrinter &printer,
|
|
mlir::Operation *op,
|
|
llvm::ArrayRef<int64_t> value);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveIntProperty
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
bool parseIntProperty(mlir::OpAsmParser &parser, int64_t &value);
|
|
|
|
void printIntProperty(mlir::OpAsmPrinter &printer, mlir::Operation *op,
|
|
int64_t value);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveSumProperty
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
bool parseSumProperty(mlir::OpAsmParser &parser, int64_t &second,
|
|
int64_t first);
|
|
|
|
void printSumProperty(mlir::OpAsmPrinter &printer, mlir::Operation *op,
|
|
int64_t second, int64_t first);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveOptionalCustomParser
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::OptionalParseResult parseOptionalCustomParser(mlir::AsmParser &p,
|
|
mlir::IntegerAttr &result);
|
|
|
|
void printOptionalCustomParser(mlir::AsmPrinter &p, mlir::Operation *,
|
|
mlir::IntegerAttr result);
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CustomDirectiveAttrElideType
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
mlir::ParseResult parseAttrElideType(mlir::AsmParser &parser,
|
|
mlir::TypeAttr type,
|
|
mlir::Attribute &attr);
|
|
|
|
void printAttrElideType(mlir::AsmPrinter &printer, mlir::Operation *op,
|
|
mlir::TypeAttr type, mlir::Attribute attr);
|
|
|
|
} // end namespace test
|
|
|
|
#endif // MLIR_TESTFORMATUTILS_H
|