SPIRVOps.cpp File Reference

#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
#include "SPIRVOpUtils.h"
#include "SPIRVParsingUtils.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVAttributes.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVEnums.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVOpTraits.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
#include "mlir/Dialect/SPIRV/IR/TargetAndABI.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/IR/Operation.h"
#include "mlir/IR/TypeUtilities.h"
#include "mlir/Interfaces/FunctionImplementation.h"
#include "llvm/ADT/APFloat.h"
#include "llvm/ADT/APInt.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/StringExtras.h"
#include "llvm/ADT/TypeSwitch.h"
#include "llvm/Support/InterleavedRange.h"
#include <cassert>
#include <numeric>
#include <optional>

Go to the source code of this file.

Functions
static ParseResult	parseOneResultSameOperandTypeOp (OpAsmParser &parser, OperationState &result)
static void	printOneResultOp (Operation *op, OpAsmPrinter &p)
template<typename BlockReadWriteOpTy>
static LogicalResult	verifyBlockReadWritePtrAndValTypes (BlockReadWriteOpTy op, Value ptr, Value val)
static Type	getElementType (Type type, ArrayRef< int32_t > indices, function_ref< InFlightDiagnostic(StringRef)> emitErrorFn)
	Walks the given type hierarchy with the given indices, potentially down to component granularity, to select an element type.
static Type	getElementType (Type type, Attribute indices, function_ref< InFlightDiagnostic(StringRef)> emitErrorFn)
static Type	getElementType (Type type, Attribute indices, Location loc)
static Type	getElementType (Type type, Attribute indices, OpAsmParser &parser, SMLoc loc)
template<typename ExtendedBinaryOp>
static LogicalResult	verifyArithmeticExtendedBinaryOp (ExtendedBinaryOp op)
static ParseResult	parseArithmeticExtendedBinaryOp (OpAsmParser &parser, OperationState &result)
static void	printArithmeticExtendedBinaryOp (Operation *op, OpAsmPrinter &printer)
static LogicalResult	verifyShiftOp (Operation *op)
static LogicalResult	verifyConstantType (spirv::ConstantOp op, Attribute value, Type opType)
static Type	getValueType (Attribute attr)
static LogicalResult	verifyFloatIntegerBuiltin (Operation *op, Type floatType, Type integerType)

Function Documentation

getElementType() [1/4]

Type getElementType ( Type type, ArrayRef< int32_t > indices, function_ref< InFlightDiagnostic(StringRef)> emitErrorFn )

static

Walks the given type hierarchy with the given indices, potentially down to component granularity, to select an element type.

Returns null type and emits errors with the given loc on failure.

Definition at line 225 of file SPIRVOps.cpp.

References indices.

Referenced by getElementType(), getElementType(), and getElementType().

getElementType() [2/4]

Type getElementType ( Type type, Attribute indices, function_ref< InFlightDiagnostic(StringRef)> emitErrorFn )

static

Definition at line 251 of file SPIRVOps.cpp.

References getElementType(), and indices.

getElementType() [3/4]

Type getElementType ( Type type, Attribute indices, Location loc )

static

Definition at line 276 of file SPIRVOps.cpp.

References getElementType(), and indices.

getElementType() [4/4]

Type getElementType ( Type type, Attribute indices, OpAsmParser & parser, SMLoc loc )

static

Definition at line 283 of file SPIRVOps.cpp.

References mlir::AsmParser::emitError(), getElementType(), and indices.

getValueType()

Type getValueType ( Attribute attr )

static

Definition at line 831 of file SPIRVOps.cpp.

References mlir::spirv::ArrayType::get(), and getValueType().

Referenced by getValueType().

parseArithmeticExtendedBinaryOp()

ParseResult parseArithmeticExtendedBinaryOp ( OpAsmParser & parser, OperationState & result )

static

Definition at line 306 of file SPIRVOps.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::parseColon(), mlir::OpAsmParser::parseOperandList(), mlir::AsmParser::parseOptionalAttrDict(), mlir::AsmParser::parseType(), mlir::OpAsmParser::resolveOperands(), result, and success().

parseOneResultSameOperandTypeOp()

ParseResult parseOneResultSameOperandTypeOp ( OpAsmParser & parser, OperationState & result )

static

Definition at line 160 of file SPIRVOps.cpp.

References mlir::AsmParser::addTypeToList(), mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::parseColon(), mlir::AsmParser::parseColonType(), mlir::OpAsmParser::parseOperandList(), mlir::AsmParser::parseOptionalAttrDict(), mlir::AsmParser::parseOptionalLParen(), mlir::AsmParser::parseRParen(), mlir::AsmParser::parseType(), mlir::OpAsmParser::resolveOperands(), result, and success().

printArithmeticExtendedBinaryOp()

void printArithmeticExtendedBinaryOp ( Operation * op, OpAsmPrinter & printer )

static

Definition at line 330 of file SPIRVOps.cpp.

References mlir::ValueTypeRange< ValueRangeT >::front(), mlir::Operation::getAttrs(), mlir::Operation::getOperands(), mlir::Operation::getResultTypes(), mlir::OpAsmPrinter::printOperands(), and mlir::OpAsmPrinter::printOptionalAttrDict().

printOneResultOp()

void printOneResultOp ( Operation * op, OpAsmPrinter & p )

static

Definition at line 189 of file SPIRVOps.cpp.

References mlir::Operation::getAttrs(), mlir::Operation::getNumResults(), mlir::Operation::getOperands(), mlir::Operation::getOperandTypes(), mlir::Operation::getResult(), mlir::Value::getType(), mlir::OpAsmPrinter::printGenericOp(), mlir::OpAsmPrinter::printOperands(), and mlir::OpAsmPrinter::printOptionalAttrDict().

verifyArithmeticExtendedBinaryOp()

template<typename ExtendedBinaryOp>

LogicalResult verifyArithmeticExtendedBinaryOp ( ExtendedBinaryOp op )

static

Definition at line 292 of file SPIRVOps.cpp.

References success().

verifyBlockReadWritePtrAndValTypes()

template<typename BlockReadWriteOpTy>

LogicalResult verifyBlockReadWritePtrAndValTypes ( BlockReadWriteOpTy op, Value ptr, Value val )

static

Definition at line 209 of file SPIRVOps.cpp.

References mlir::Value::getType(), and success().

verifyConstantType()

LogicalResult verifyConstantType ( spirv::ConstantOp op, Attribute value, Type opType )

static

Definition at line 602 of file SPIRVOps.cpp.

References success(), and verifyConstantType().

Referenced by verifyConstantType().

verifyFloatIntegerBuiltin()

LogicalResult verifyFloatIntegerBuiltin ( Operation * op, Type floatType, Type integerType )

static

Definition at line 2123 of file SPIRVOps.cpp.

References mlir::Operation::emitOpError(), getNumElements(), and success().

verifyShiftOp()

LogicalResult verifyShiftOp ( Operation * op )

static

Definition at line 338 of file SPIRVOps.cpp.

References mlir::Operation::emitError(), mlir::Operation::getOperand(), mlir::Operation::getResult(), mlir::Value::getType(), and success().