MemRefToSPIRV.cpp File Reference

#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVEnums.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
#include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h"
#include "mlir/IR/BuiltinTypes.h"
#include "llvm/Support/Debug.h"
#include <optional>

Go to the source code of this file.

Namespaces
	mlir
	Include the generated interface declarations.

Macros
#define	DEBUG_TYPE   "memref-to-spirv-pattern"
#define	ATOMIC_CASE(kind, spirvOp)

Functions
static Value	getOffsetForBitwidth (Location loc, Value srcIdx, int sourceBits, int targetBits, OpBuilder &builder)
	Returns the offset of the value in targetBits representation. More...
static Value	adjustAccessChainForBitwidth (const SPIRVTypeConverter &typeConverter, spirv::AccessChainOp op, int sourceBits, int targetBits, OpBuilder &builder)
	Returns an adjusted spirv::AccessChainOp. More...
static Value	castBoolToIntN (Location loc, Value srcBool, Type dstType, OpBuilder &builder)
	Casts the given srcBool into an integer of dstType. More...
static Value	shiftValue (Location loc, Value value, Value offset, Value mask, OpBuilder &builder)
	Returns the targetBits-bit value shifted by the given offset, and cast to the type destination type, and masked. More...
static bool	isAllocationSupported (Operation *allocOp, MemRefType type)
	Returns true if the allocations of memref type generated from allocOp can be lowered to SPIR-V. More...
static std::optional< spirv::Scope >	getAtomicOpScope (MemRefType type)
	Returns the scope to use for atomic operations use for emulating store operations of unsupported integer bitwidths, based on the memref type. More...
static Value	castIntNToBool (Location loc, Value srcInt, OpBuilder &builder)
	Casts the given srcInt into a boolean value. More...
void	mlir::populateMemRefToSPIRVPatterns (SPIRVTypeConverter &typeConverter, RewritePatternSet &patterns)
	Appends to a pattern list additional patterns for translating MemRef ops to SPIR-V ops. More...

Macro Definition Documentation

ATOMIC_CASE

#define ATOMIC_CASE	(		kind,
			spirvOp
	)

Value:

  case arith::AtomicRMWKind::kind:                                             \
    rewriter.replaceOpWithNewOp<spirv::spirvOp>(                               \
        atomicOp, resultType, ptr, *scope,                                     \
        spirv::MemorySemantics::AcquireRelease, adaptor.getValue());           \
    break

DEBUG_TYPE

#define DEBUG_TYPE   "memref-to-spirv-pattern"

Definition at line 23 of file MemRefToSPIRV.cpp.

Function Documentation

adjustAccessChainForBitwidth()

static Value adjustAccessChainForBitwidth ( const SPIRVTypeConverter &  typeConverter, spirv::AccessChainOp  op, int  sourceBits, int  targetBits, OpBuilder &  builder  )

static

Returns an adjusted spirv::AccessChainOp.

Based on the extension/capabilities, certain integer bitwidths sourceBits might not be supported. During conversion if a memref of an unsupported type is used, load/stores to this memref need to be modified to use a supported higher bitwidth targetBits and extracting the required bits. For an accessing a 1D array (spirv.array or spirv.rtarray), the last index is modified to load the bits needed. The extraction of the actual bits needed are handled separately. Note that this only works for a 1-D tensor.

Definition at line 63 of file MemRefToSPIRV.cpp.

castBoolToIntN()

static Value castBoolToIntN ( Location  loc, Value  srcBool, Type  dstType, OpBuilder &  builder  )

static

Casts the given srcBool into an integer of dstType.

Definition at line 81 of file MemRefToSPIRV.cpp.

References mlir::OpBuilder::create(), mlir::Value::getType(), getZero(), and mlir::Type::isInteger().

Referenced by shiftValue().

castIntNToBool()

static Value castIntNToBool ( Location  loc, Value  srcInt, OpBuilder &  builder  )

static

Casts the given srcInt into a boolean value.

Definition at line 157 of file MemRefToSPIRV.cpp.

References mlir::OpBuilder::create(), mlir::Value::getType(), and mlir::Type::isInteger().

getAtomicOpScope()

static std::optional<spirv::Scope> getAtomicOpScope ( MemRefType  type )

static

Returns the scope to use for atomic operations use for emulating store operations of unsupported integer bitwidths, based on the memref type.

Returns std::nullopt on failure.

Definition at line 143 of file MemRefToSPIRV.cpp.

getOffsetForBitwidth()

static Value getOffsetForBitwidth ( Location  loc, Value  srcIdx, int  sourceBits, int  targetBits, OpBuilder &  builder  )

static

Returns the offset of the value in targetBits representation.

srcIdx is an index into a 1-D array with each element having sourceBits. It's assumed to be non-negative.

When accessing an element in the array treating as having elements of targetBits, multiple values are loaded in the same time. The method returns the offset where the srcIdx locates in the value. For example, if sourceBits equals to 8 and targetBits equals to 32, the x-th element is located at (x % 4) * 8. Because there are four elements in one i32, and one element has 8 bits.

Definition at line 42 of file MemRefToSPIRV.cpp.

References mlir::OpBuilder::create(), mlir::Builder::getIntegerAttr(), and mlir::Value::getType().

isAllocationSupported()

static bool isAllocationSupported ( Operation *  allocOp, MemRefType  type  )

static

Returns true if the allocations of memref type generated from allocOp can be lowered to SPIR-V.

Definition at line 116 of file MemRefToSPIRV.cpp.

References mlir::Type::isIntOrFloat().

shiftValue()

static Value shiftValue ( Location  loc, Value  value, Value  offset, Value  mask, OpBuilder &  builder  )

static

Returns the targetBits-bit value shifted by the given offset, and cast to the type destination type, and masked.

Definition at line 93 of file MemRefToSPIRV.cpp.

References castBoolToIntN(), mlir::OpBuilder::create(), mlir::Builder::getIntegerType(), mlir::Type::getIntOrFloatBitWidth(), and mlir::Value::getType().