SPIR-V struct type. More...

#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"

Inheritance diagram for mlir::spirv::StructType:

Collaboration diagram for mlir::spirv::StructType:

Classes
class	ElementTypeRange
	Range class for element types. More...

struct	MemberDecorationInfo

Public Types
using	OffsetInfo = uint32_t

Public Types inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
using	Base = StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits... >
	Utility declarations for the concrete attribute class. More...

using	ImplType = StorageT

using	HasTraitFn = bool(*)(TypeID)

Public Member Functions
StringRef	getIdentifier () const
	For literal structs, return an empty string. More...

bool	isIdentified () const
	Returns true if the StructType is identified. More...

unsigned	getNumElements () const

Type	getElementType (unsigned) const

ElementTypeRange	getElementTypes () const

bool	hasOffset () const

uint64_t	getMemberOffset (unsigned) const

void	getMemberDecorations (SmallVectorImpl< StructType::MemberDecorationInfo > &memberDecorations) const

void	getMemberDecorations (unsigned i, SmallVectorImpl< StructType::MemberDecorationInfo > &decorationsInfo) const

LogicalResult	trySetBody (ArrayRef< Type > memberTypes, ArrayRef< OffsetInfo > offsetInfo={}, ArrayRef< MemberDecorationInfo > memberDecorations={})
	Sets the contents of an incomplete identified StructType. More...

void	getExtensions (SPIRVType::ExtensionArrayRefVector &extensions, Optional< StorageClass > storage=llvm::None)

void	getCapabilities (SPIRVType::CapabilityArrayRefVector &capabilities, Optional< StorageClass > storage=llvm::None)

Static Public Member Functions
static StructType	get (ArrayRef< Type > memberTypes, ArrayRef< OffsetInfo > offsetInfo={}, ArrayRef< MemberDecorationInfo > memberDecorations={})
	Construct a literal StructType with at least one member. More...

static StructType	getIdentified (MLIRContext *context, StringRef identifier)
	Construct an identified StructType. More...

static StructType	getEmpty (MLIRContext *context, StringRef identifier="")
	Construct a (possibly identified) StructType with no members. More...

Static Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
static TypeID	getTypeID ()
	Return a unique identifier for the concrete type. More...

template<typename T >
static bool	classof (T val)
	Provide an implementation of 'classof' that compares the type id of the provided value with that of the concrete type. More...

static detail::InterfaceMap	getInterfaceMap ()
	Returns an interface map for the interfaces registered to this storage user. More...

static HasTraitFn	getHasTraitFn ()
	Returns the function that returns true if the given Trait ID matches the IDs of any of the traits defined by the storage user. More...

template<typename... IfaceModels>
static void	attachInterface (MLIRContext &context)
	Attach the given models as implementations of the corresponding interfaces for the concrete storage user class. More...

template<typename... Args>
static ConcreteT	get (MLIRContext *ctx, Args... args)
	Get or create a new ConcreteT instance within the ctx. More...

template<typename... Args>
static ConcreteT	getChecked (const Location &loc, Args... args)
	Get or create a new ConcreteT instance within the ctx, defined at the given, potentially unknown, location. More...

template<typename... Args>
static ConcreteT	getChecked (function_ref< InFlightDiagnostic()> emitErrorFn, MLIRContext *ctx, Args... args)
	Get or create a new ConcreteT instance within the ctx. More...

static ConcreteT	getFromOpaquePointer (const void *ptr)
	Get an instance of the concrete type from a void pointer. More...

Additional Inherited Members
Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
template<typename... Args>
LogicalResult	mutate (Args &&...args)
	Mutate the current storage instance. More...

ImplType *	getImpl () const
	Utility for easy access to the storage instance. More...

Static Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
template<typename... Args>
static LogicalResult	verify (Args... args)
	Default implementation that just returns success. More...

Detailed Description

SPIR-V struct type.

Two kinds of struct types are supported:

Literal: a literal struct type is uniqued by its fields (types + offset info + decoration info).
Identified: an indentified struct type is uniqued by its string identifier (name). This is useful in representing recursive structs. For example, the following C struct:

struct A { A* next; };

would be represented in MLIR as:

!spirv.struct<A, (!spirv.ptr<!spirv.struct, Generic>)>

In the above, expressing recursive struct types is accomplished by giving a recursive struct a unique identified and using that identifier in the struct definition for recursive references.

Definition at line 279 of file SPIRVTypes.h.

Member Typedef Documentation

◆ OffsetInfo

using mlir::spirv::StructType::OffsetInfo = uint32_t

Definition at line 286 of file SPIRVTypes.h.

Member Function Documentation

◆ get()

StructType StructType::get	(	ArrayRef< Type >	memberTypes,
		ArrayRef< OffsetInfo >	offsetInfo = `{}`,
		ArrayRef< MemberDecorationInfo >	memberDecorations = `{}`
	)

static

Construct a literal StructType with at least one member.

Definition at line 1037 of file SPIRVTypes.cpp.

Referenced by mlir::spirv::Deserializer::collect(), createGlobalVarForEntryPointArgument(), mlir::VulkanLayoutUtils::decorateType(), getPushConstantStorageType(), parseStructType(), and wrapInStructAndGetPointer().

◆ getCapabilities()

void StructType::getCapabilities	(	SPIRVType::CapabilityArrayRefVector &	capabilities,
		Optional< StorageClass >	storage = `llvm::None`
	)

Definition at line 1136 of file SPIRVTypes.cpp.

◆ getElementType()

Type StructType::getElementType ( unsigned index ) const

Definition at line 1080 of file SPIRVTypes.cpp.

References getNumElements().

Referenced by mlir::VulkanLayoutUtils::decorateType(), mlir::spirv::CompositeType::getElementType(), getRuntimeArrayElementType(), print(), and verifyJointMatrixMad().

◆ getElementTypes()

StructType::ElementTypeRange StructType::getElementTypes ( ) const

Definition at line 1085 of file SPIRVTypes.cpp.

References getNumElements().

Referenced by convertStructTypePacked(), and convertStructTypeWithOffset().

◆ getEmpty()

StructType StructType::getEmpty	(	MLIRContext *	context,
		StringRef	identifier = `""`
	)

static

Construct a (possibly identified) StructType with no members.

Note: this method might fail in a multi-threaded setup if another thread created an identified struct with the same identifier but with different contents before returning. In which case, an empty (default-constructed) StructType is returned.

Definition at line 1060 of file SPIRVTypes.cpp.

References mlir::failed(), isIdentified(), and trySetBody().

Referenced by mlir::spirv::Deserializer::collect(), and parseStructType().

◆ getExtensions()

void StructType::getExtensions	(	SPIRVType::ExtensionArrayRefVector &	extensions,
		Optional< StorageClass >	storage = `llvm::None`
	)

Definition at line 1130 of file SPIRVTypes.cpp.

◆ getIdentified()

StructType StructType::getIdentified	(	MLIRContext *	context,
		StringRef	identifier
	)

static

Construct an identified StructType.

This creates a StructType whose body (member types, offset info, and decorations) is not set yet. A call to StructType::trySetBody(...) must follow when the StructType contents are available (e.g. parsed or deserialized).

Note: If another thread creates (or had already created) a struct with the same identifier, that struct will be returned as a result.

Definition at line 1050 of file SPIRVTypes.cpp.

Referenced by mlir::spirv::Deserializer::collect(), parseStructType(), and mlir::spirv::Serializer::processTypeDecoration< spirv::RuntimeArrayType >().

◆ getIdentifier()

StringRef StructType::getIdentifier ( ) const

For literal structs, return an empty string.

For identified structs, return the struct's identifier.

Definition at line 1074 of file SPIRVTypes.cpp.

Referenced by print(), and mlir::spirv::Serializer::processTypeDecoration< spirv::RuntimeArrayType >().

◆ getMemberDecorations() [1/2]

void StructType::getMemberDecorations ( SmallVectorImpl< StructType::MemberDecorationInfo > & memberDecorations ) const

Definition at line 1097 of file SPIRVTypes.cpp.

Referenced by convertStructType(), mlir::VulkanLayoutUtils::decorateType(), and print().

◆ getMemberDecorations() [2/2]

void StructType::getMemberDecorations	(	unsigned	i,
		SmallVectorImpl< StructType::MemberDecorationInfo > &	decorationsInfo
	)		const

Definition at line 1106 of file SPIRVTypes.cpp.

References getNumElements().

◆ getMemberOffset()

uint64_t StructType::getMemberOffset ( unsigned index ) const

Definition at line 1092 of file SPIRVTypes.cpp.

References getNumElements().

Referenced by print().

◆ getNumElements()

unsigned StructType::getNumElements ( ) const

Definition at line 1078 of file SPIRVTypes.cpp.

Referenced by mlir::VulkanLayoutUtils::decorateType(), mlir::VulkanLayoutUtils::isLegalType(), print(), and verifyJointMatrixMad().

◆ hasOffset()

bool StructType::hasOffset ( ) const

Definition at line 1090 of file SPIRVTypes.cpp.

Referenced by convertStructType(), mlir::VulkanLayoutUtils::isLegalType(), and print().

◆ isIdentified()

bool StructType::isIdentified ( ) const

Returns true if the StructType is identified.

Definition at line 1076 of file SPIRVTypes.cpp.

Referenced by mlir::VulkanLayoutUtils::decorateType(), getEmpty(), print(), and mlir::spirv::Serializer::processTypeDecoration< spirv::RuntimeArrayType >().

◆ trySetBody()

LogicalResult StructType::trySetBody	(	ArrayRef< Type >	memberTypes,
		ArrayRef< OffsetInfo >	offsetInfo = `{}`,
		ArrayRef< MemberDecorationInfo >	memberDecorations = `{}`
	)

Sets the contents of an incomplete identified StructType.

This method must be called only for identified StructTypes and it must be called only once per instance. Otherwise, failure() is returned.

Definition at line 1124 of file SPIRVTypes.cpp.

Referenced by getEmpty().

The documentation for this class was generated from the following files:

include/mlir/Dialect/SPIRV/IR/SPIRVTypes.h
lib/Dialect/SPIRV/IR/SPIRVTypes.cpp

Classes

Public Types

Public Member Functions

Static Public Member Functions

Additional Inherited Members

Detailed Description

Member Typedef Documentation

◆ OffsetInfo

Member Function Documentation

◆ get()

◆ getCapabilities()

◆ getElementType()

◆ getElementTypes()

◆ getEmpty()

◆ getExtensions()

◆ getIdentified()

◆ getIdentifier()

◆ getMemberDecorations() [1/2]

◆ getMemberDecorations() [2/2]

◆ getMemberOffset()

◆ getNumElements()

◆ hasOffset()

◆ isIdentified()

◆ trySetBody()