SPIRVDialect.cpp File Reference

#include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h"
#include "SPIRVParsingUtils.h"
#include "mlir/Dialect/GPU/IR/CompilationInterfaces.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
#include "mlir/Dialect/SPIRV/IR/TargetAndABI.h"
#include "mlir/Dialect/UB/IR/UBOps.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/DialectImplementation.h"
#include "mlir/IR/MLIRContext.h"
#include "mlir/Parser/Parser.h"
#include "mlir/Transforms/InliningUtils.h"
#include "llvm/ADT/Sequence.h"
#include "llvm/ADT/StringExtras.h"
#include "llvm/ADT/TypeSwitch.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVOpsDialect.cpp.inc"
#include "mlir/Dialect/SPIRV/IR/SPIRVOps.cpp.inc"

Go to the source code of this file.

Macros
#define	GET_OP_LIST

Functions
static bool	containsReturn (Region &region)
	Returns true if the given region contains spirv.Return or spirv.ReturnValue ops.
template<typename ValTy>
static std::optional< ValTy >	parseAndVerify (SPIRVDialect const &dialect, DialectAsmParser &parser)
template<>
std::optional< Type >	parseAndVerify< Type > (SPIRVDialect const &dialect, DialectAsmParser &parser)
template<>
std::optional< unsigned >	parseAndVerify< unsigned > (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseAndVerifyType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseAndVerifyMatrixType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseAndVerifySampledImageType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static LogicalResult	parseOptionalArrayStride (const SPIRVDialect &dialect, DialectAsmParser &parser, unsigned &stride)
	Parses an optional , stride = N assembly segment.
static Type	parseArrayType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseCooperativeMatrixType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseTensorArmType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parsePointerType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseRuntimeArrayType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseMatrixType (SPIRVDialect const &dialect, DialectAsmParser &parser)
template<>
std::optional< Type >	parseAndVerify< Type > (SPIRVDialect const &dialect, DialectAsmParser &parser)
template<typename IntTy>
static std::optional< IntTy >	parseAndVerifyInteger (SPIRVDialect const &dialect, DialectAsmParser &parser)
template<>
std::optional< unsigned >	parseAndVerify< unsigned > (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseImageType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static Type	parseSampledImageType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static ParseResult	parseStructMemberDecorations (SPIRVDialect const &dialect, DialectAsmParser &parser, ArrayRef< Type > memberTypes, SmallVectorImpl< StructType::OffsetInfo > &offsetInfo, SmallVectorImpl< StructType::MemberDecorationInfo > &memberDecorationInfo)
static Type	parseStructType (SPIRVDialect const &dialect, DialectAsmParser &parser)
static void	print (ArrayType type, DialectAsmPrinter &os)
static void	print (RuntimeArrayType type, DialectAsmPrinter &os)
static void	print (PointerType type, DialectAsmPrinter &os)
static void	print (ImageType type, DialectAsmPrinter &os)
static void	print (SampledImageType type, DialectAsmPrinter &os)
static void	print (StructType type, DialectAsmPrinter &os)
static void	print (CooperativeMatrixType type, DialectAsmPrinter &os)
static void	print (MatrixType type, DialectAsmPrinter &os)
static void	print (TensorArmType type, DialectAsmPrinter &os)
static LogicalResult	verifyRegionAttribute (Location loc, Type valueType, NamedAttribute attribute)
	Verifies the given SPIR-V attribute attached to a value of the given valueType is valid.

Macro Definition Documentation

GET_OP_LIST

#define GET_OP_LIST

Function Documentation

containsReturn()

bool containsReturn ( Region & region )

inlinestatic

Returns true if the given region contains spirv.Return or spirv.ReturnValue ops.

Definition at line 43 of file SPIRVDialect.cpp.

References mlir::Block::getTerminator().

parseAndVerify()

template<typename ValTy>

std::optional< ValTy > parseAndVerify ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 516 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), and mlir::AsmParser::parseKeyword().

Referenced by parseStructMemberDecorations(), and parseStructType().

parseAndVerify< Type >() [1/2]

template<>

std::optional< Type > parseAndVerify< Type >	(	SPIRVDialect const &	dialect,
		DialectAsmParser &	parser )

Definition at line 516 of file SPIRVDialect.cpp.

parseAndVerify< Type >() [2/2]

template<>

std::optional< Type > parseAndVerify< Type >	(	SPIRVDialect const &	dialect,
		DialectAsmParser &	parser )

Definition at line 516 of file SPIRVDialect.cpp.

parseAndVerify< unsigned >() [1/2]

template<>

std::optional< unsigned > parseAndVerify< unsigned >	(	SPIRVDialect const &	dialect,
		DialectAsmParser &	parser )

Definition at line 541 of file SPIRVDialect.cpp.

References mlir::AsmParser::parseInteger().

Referenced by parseOptionalArrayStride().

parseAndVerify< unsigned >() [2/2]

template<>

std::optional< unsigned > parseAndVerify< unsigned >	(	SPIRVDialect const &	dialect,
		DialectAsmParser &	parser )

Definition at line 541 of file SPIRVDialect.cpp.

References mlir::AsmParser::parseInteger().

Referenced by parseOptionalArrayStride().

parseAndVerifyInteger()

template<typename IntTy>

std::optional< IntTy > parseAndVerifyInteger ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 541 of file SPIRVDialect.cpp.

parseAndVerifyMatrixType()

Type parseAndVerifyMatrixType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 211 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), and mlir::AsmParser::parseType().

Referenced by parseMatrixType().

parseAndVerifySampledImageType()

Type parseAndVerifySampledImageType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 247 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), and mlir::AsmParser::parseType().

Referenced by parseSampledImageType().

parseAndVerifyType()

Type parseAndVerifyType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 163 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), mlir::Type::getDialect(), mlir::spirv::ScalarType::isValid(), and mlir::AsmParser::parseType().

Referenced by parseArrayType(), parseCooperativeMatrixType(), parsePointerType(), parseRuntimeArrayType(), and parseTensorArmType().

parseArrayType()

Type parseArrayType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 305 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::spirv::ArrayType::get(), mlir::AsmParser::getCurrentLocation(), parseAndVerifyType(), mlir::AsmParser::parseDimensionList(), mlir::AsmParser::parseGreater(), mlir::AsmParser::parseLess(), and parseOptionalArrayStride().

parseCooperativeMatrixType()

Type parseCooperativeMatrixType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 344 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::spirv::CooperativeMatrixType::get(), mlir::AsmParser::getCurrentLocation(), parseAndVerifyType(), mlir::AsmParser::parseComma(), mlir::AsmParser::parseDimensionList(), mlir::spirv::parseEnumKeywordAttr(), mlir::AsmParser::parseGreater(), and mlir::AsmParser::parseLess().

parseImageType()

Type parseImageType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 607 of file SPIRVDialect.cpp.

References mlir::spirv::ImageType::get(), mlir::AsmParser::parseGreater(), and mlir::AsmParser::parseLess().

parseMatrixType()

Type parseMatrixType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 480 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::spirv::MatrixType::get(), mlir::AsmParser::getCurrentLocation(), parseAndVerifyMatrixType(), mlir::AsmParser::parseDimensionList(), mlir::AsmParser::parseGreater(), and mlir::AsmParser::parseLess().

parseOptionalArrayStride()

LogicalResult parseOptionalArrayStride ( const SPIRVDialect & dialect, DialectAsmParser & parser, unsigned & stride )

static

Parses an optional , stride = N assembly segment.

If no parsing failure occurs, writes N to stride if existing and writes 0 to stride if missing.

Definition at line 275 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), parseAndVerify< unsigned >(), mlir::AsmParser::parseEqual(), mlir::AsmParser::parseKeyword(), mlir::AsmParser::parseOptionalComma(), and success().

Referenced by parseArrayType(), and parseRuntimeArrayType().

parsePointerType()

Type parsePointerType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 434 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::spirv::PointerType::get(), mlir::AsmParser::getCurrentLocation(), parseAndVerifyType(), mlir::AsmParser::parseComma(), mlir::AsmParser::parseGreater(), mlir::AsmParser::parseKeyword(), and mlir::AsmParser::parseLess().

parseRuntimeArrayType()

Type parseRuntimeArrayType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 461 of file SPIRVDialect.cpp.

References mlir::spirv::RuntimeArrayType::get(), parseAndVerifyType(), mlir::AsmParser::parseGreater(), mlir::AsmParser::parseLess(), and parseOptionalArrayStride().

parseSampledImageType()

Type parseSampledImageType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 625 of file SPIRVDialect.cpp.

References mlir::spirv::SampledImageType::get(), parseAndVerifySampledImageType(), mlir::AsmParser::parseGreater(), and mlir::AsmParser::parseLess().

parseStructMemberDecorations()

ParseResult parseStructMemberDecorations ( SPIRVDialect const & dialect, DialectAsmParser & parser, ArrayRef< Type > memberTypes, SmallVectorImpl< StructType::OffsetInfo > & offsetInfo, SmallVectorImpl< StructType::MemberDecorationInfo > & memberDecorationInfo )

static

Definition at line 640 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), mlir::OptionalParseResult::has_value(), parseAndVerify(), mlir::AsmParser::parseAttribute(), mlir::AsmParser::parseComma(), mlir::AsmParser::parseCommaSeparatedList(), mlir::AsmParser::parseOptionalEqual(), mlir::AsmParser::parseOptionalInteger(), mlir::AsmParser::parseOptionalRSquare(), mlir::AsmParser::parseRSquare(), and success().

Referenced by parseStructType().

parseStructType()

Type parseStructType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 707 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::spirv::StructType::get(), mlir::spirv::StructType::getEmpty(), mlir::spirv::StructType::getIdentified(), mlir::AsmParser::getNameLoc(), parseAndVerify(), mlir::AsmParser::parseAttribute(), mlir::AsmParser::parseComma(), mlir::AsmParser::parseGreater(), mlir::AsmParser::parseLess(), mlir::AsmParser::parseLParen(), mlir::AsmParser::parseOptionalComma(), mlir::AsmParser::parseOptionalEqual(), mlir::AsmParser::parseOptionalGreater(), mlir::AsmParser::parseOptionalKeyword(), mlir::AsmParser::parseOptionalLSquare(), mlir::AsmParser::parseOptionalRParen(), mlir::AsmParser::parseRParen(), parseStructMemberDecorations(), mlir::AsmParser::parseType(), success(), mlir::spirv::StructType::trySetBody(), and mlir::AsmParser::tryStartCyclicParse().

parseTensorArmType()

Type parseTensorArmType ( SPIRVDialect const & dialect, DialectAsmParser & parser )

static

Definition at line 381 of file SPIRVDialect.cpp.

References mlir::AsmParser::emitError(), mlir::spirv::TensorArmType::get(), mlir::AsmParser::getCurrentLocation(), parseAndVerifyType(), mlir::AsmParser::parseDimensionList(), mlir::AsmParser::parseGreater(), mlir::AsmParser::parseLess(), mlir::AsmParser::parseOptionalStar(), and mlir::AsmParser::parseXInDimensionList().

print() [1/9]

void print ( ArrayType type, DialectAsmPrinter & os )

static

Definition at line 862 of file SPIRVDialect.cpp.

References mlir::spirv::ArrayType::getArrayStride(), mlir::spirv::ArrayType::getElementType(), and mlir::spirv::ArrayType::getNumElements().

print() [2/9]

void print ( CooperativeMatrixType type, DialectAsmPrinter & os )

static

Definition at line 956 of file SPIRVDialect.cpp.

References mlir::spirv::CooperativeMatrixType::getColumns(), mlir::spirv::CooperativeMatrixType::getElementType(), mlir::spirv::CooperativeMatrixType::getRows(), mlir::spirv::CooperativeMatrixType::getScope(), and mlir::spirv::CooperativeMatrixType::getUse().

print() [3/9]

void print ( ImageType type, DialectAsmPrinter & os )

static

Definition at line 881 of file SPIRVDialect.cpp.

References mlir::spirv::ImageType::getArrayedInfo(), mlir::spirv::ImageType::getDepthInfo(), mlir::spirv::ImageType::getDim(), mlir::spirv::ImageType::getElementType(), mlir::spirv::ImageType::getImageFormat(), mlir::spirv::ImageType::getSamplerUseInfo(), and mlir::spirv::ImageType::getSamplingInfo().

print() [4/9]

void print ( MatrixType type, DialectAsmPrinter & os )

static

Definition at line 962 of file SPIRVDialect.cpp.

References mlir::spirv::MatrixType::getColumnType(), and mlir::spirv::MatrixType::getNumColumns().

print() [5/9]

void print ( PointerType type, DialectAsmPrinter & os )

static

Definition at line 876 of file SPIRVDialect.cpp.

References mlir::spirv::PointerType::getPointeeType(), and mlir::spirv::PointerType::getStorageClass().

print() [6/9]

void print ( RuntimeArrayType type, DialectAsmPrinter & os )

static

Definition at line 869 of file SPIRVDialect.cpp.

References mlir::spirv::RuntimeArrayType::getArrayStride(), and mlir::spirv::RuntimeArrayType::getElementType().

print() [7/9]

void print ( SampledImageType type, DialectAsmPrinter & os )

static

Definition at line 890 of file SPIRVDialect.cpp.

References mlir::spirv::SampledImageType::getImageType().

print() [8/9]

void print ( StructType type, DialectAsmPrinter & os )

static

Definition at line 894 of file SPIRVDialect.cpp.

References mlir::spirv::StructType::getElementType(), mlir::spirv::StructType::getIdentifier(), mlir::spirv::StructType::getMemberDecorations(), mlir::spirv::StructType::getMemberOffset(), mlir::spirv::StructType::getNumElements(), mlir::spirv::StructType::getStructDecorations(), mlir::spirv::StructType::hasOffset(), mlir::spirv::StructType::isIdentified(), mlir::AsmPrinter::printAttributeWithoutType(), and mlir::AsmPrinter::tryStartCyclicPrint().

print() [9/9]

void print ( TensorArmType type, DialectAsmPrinter & os )

static

Definition at line 967 of file SPIRVDialect.cpp.

References mlir::spirv::TensorArmType::getElementType(), mlir::spirv::TensorArmType::getShape(), and mlir::spirv::TensorArmType::hasRank().

verifyRegionAttribute()

LogicalResult verifyRegionAttribute ( Location loc, Type valueType, NamedAttribute attribute )

static

Verifies the given SPIR-V attribute attached to a value of the given valueType is valid.

Definition at line 1036 of file SPIRVDialect.cpp.

References mlir::emitError(), mlir::spirv::getInterfaceVarABIAttrName(), mlir::NamedAttribute::getName(), mlir::NamedAttribute::getValue(), mlir::Type::isIntOrIndexOrFloat(), and success().