GPUOpsLowering.cpp File Reference

#include "GPUOpsLowering.h"
#include "mlir/Conversion/GPUCommon/GPUCommonPass.h"
#include "mlir/Conversion/LLVMCommon/VectorPattern.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/IR/Attributes.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinTypes.h"
#include "llvm/ADT/SmallVectorExtras.h"
#include "llvm/ADT/StringSet.h"
#include "llvm/Support/FormatVariadic.h"

Go to the source code of this file.

Functions
static SmallString< 16 >	getUniqueSymbolName (Operation *moduleOp, StringRef prefix)
static Value	scalarizeVectorOpHelper (Operation *op, ValueRange operands, Type llvm1DVectorTy, ConversionPatternRewriter &rewriter, const LLVMTypeConverter &converter)
	Helper for impl::scalarizeVectorOp.
static IntegerAttr	wrapNumericMemorySpace (MLIRContext *ctx, unsigned space)
static LLVM::GlobalOp	getDynamicSharedMemorySymbol (ConversionPatternRewriter &rewriter, gpu::GPUModuleOp moduleOp, gpu::DynamicSharedMemoryOp op, const LLVMTypeConverter *typeConverter, MemRefType memrefType, unsigned alignmentBit)
	Generates a symbol with 0-sized array type for dynamic shared memory usage, or uses existing symbol.

Function Documentation

getDynamicSharedMemorySymbol()

LLVM::GlobalOp getDynamicSharedMemorySymbol ( ConversionPatternRewriter & rewriter, gpu::GPUModuleOp moduleOp, gpu::DynamicSharedMemoryOp op, const LLVMTypeConverter * typeConverter, MemRefType memrefType, unsigned alignmentBit )

static

Generates a symbol with 0-sized array type for dynamic shared memory usage, or uses existing symbol.

Definition at line 670 of file GPUOpsLowering.cpp.

References mlir::SymbolTable::generateSymbolName(), and mlir::LLVMTypeConverter::getMemRefAddressSpace().

Referenced by mlir::GPUDynamicSharedMemoryOpLowering::matchAndRewrite().

getUniqueSymbolName()

SmallString< 16 > getUniqueSymbolName ( Operation * moduleOp, StringRef prefix )

static

Definition at line 36 of file GPUOpsLowering.cpp.

References mlir::SymbolTable::lookupSymbolIn().

Referenced by mlir::getOrCreateStringConstant().

scalarizeVectorOpHelper()

Value scalarizeVectorOpHelper ( Operation * op, ValueRange operands, Type llvm1DVectorTy, ConversionPatternRewriter & rewriter, const LLVMTypeConverter & converter )

static

Helper for impl::scalarizeVectorOp.

Scalarizes vectors to elements. Used either directly (for ops on 1D vectors) or as the callback passed to detail::handleMultidimensionalVectors (for ops on higher-rank vectors).

Definition at line 610 of file GPUOpsLowering.cpp.

References mlir::Operation::getAttrs(), mlir::OperationName::getIdentifier(), mlir::Operation::getLoc(), mlir::Operation::getName(), mlir::Operation::getResult(), and result.

Referenced by mlir::impl::scalarizeVectorOp().

wrapNumericMemorySpace()

IntegerAttr wrapNumericMemorySpace ( MLIRContext * ctx, unsigned space )

static

Definition at line 664 of file GPUOpsLowering.cpp.

Referenced by mlir::populateGpuMemorySpaceAttributeConversions().