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

using	Base = StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits... >
	Utility declarations for the concrete attribute class. More...

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, std::optional< StorageClass > storage=std::nullopt)

void	getCapabilities (SPIRVType::CapabilityArrayRefVector &capabilities, std::optional< StorageClass > storage=std::nullopt)

Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
ImplType *	getImpl () const
	Utility for easy access to the storage instance. More...

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...

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 280 of file SPIRVTypes.h.

Member Typedef Documentation

◆ Base

using mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::Base = StorageUserBase<ConcreteT, BaseT, StorageT, UniquerT, Traits...>

Utility declarations for the concrete attribute class.

Definition at line 99 of file StorageUniquerSupport.h.

◆ OffsetInfo

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

Definition at line 287 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 1038 of file SPIRVTypes.cpp.

Referenced by createGlobalVarForEntryPointArgument(), getPushConstantStorageType(), parseStructType(), and wrapInStructAndGetPointer().

◆ getCapabilities()

void StructType::getCapabilities	(	SPIRVType::CapabilityArrayRefVector &	capabilities,
		std::optional< StorageClass >	storage = `std::nullopt`
	)

Definition at line 1137 of file SPIRVTypes.cpp.

References getElementTypes().

◆ getElementType()

Type StructType::getElementType ( unsigned index ) const

Definition at line 1081 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl(), and getNumElements().

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

◆ getElementTypes()

StructType::ElementTypeRange StructType::getElementTypes ( ) const

Definition at line 1086 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl(), and getNumElements().

Referenced by convertStructTypePacked(), convertStructTypeWithOffset(), getCapabilities(), and getExtensions().

◆ 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 1061 of file SPIRVTypes.cpp.

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

Referenced by parseStructType().

◆ getExtensions()

void StructType::getExtensions	(	SPIRVType::ExtensionArrayRefVector &	extensions,
		std::optional< StorageClass >	storage = `std::nullopt`
	)

Definition at line 1131 of file SPIRVTypes.cpp.

References getElementTypes().

◆ 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 1051 of file SPIRVTypes.cpp.

Referenced by parseStructType().

◆ getIdentifier()

StringRef StructType::getIdentifier ( ) const

For literal structs, return an empty string.

For identified structs, return the struct's identifier.

Definition at line 1075 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by print().

◆ getMemberDecorations() [1/2]

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

Definition at line 1098 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by convertStructType(), and print().

◆ getMemberDecorations() [2/2]

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

Definition at line 1107 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl(), and getNumElements().

◆ getMemberOffset()

uint64_t StructType::getMemberOffset ( unsigned index ) const

Definition at line 1093 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl(), and getNumElements().

Referenced by print().

◆ getNumElements()

unsigned StructType::getNumElements ( ) const

Definition at line 1079 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by getElementType(), getElementTypes(), getMemberDecorations(), getMemberOffset(), mlir::VulkanLayoutUtils::isLegalType(), and print().

◆ hasOffset()

bool StructType::hasOffset ( ) const

Definition at line 1091 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

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

◆ isIdentified()

bool StructType::isIdentified ( ) const

Returns true if the StructType is identified.

Definition at line 1077 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by getEmpty(), and print().

◆ 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 1125 of file SPIRVTypes.cpp.

Referenced by getEmpty(), and parseStructType().

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

◆ Base

◆ OffsetInfo

Member Function Documentation

◆ get()

◆ getCapabilities()

◆ getElementType()

◆ getElementTypes()

◆ getEmpty()

◆ getExtensions()

◆ getIdentified()

◆ getIdentifier()

◆ getMemberDecorations() [1/2]

◆ getMemberDecorations() [2/2]

◆ getMemberOffset()

◆ getNumElements()

◆ hasOffset()

◆ isIdentified()

◆ trySetBody()