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

Function Documentation

getDynamicSharedMemorySymbol()

static 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::LLVMTypeConverter::convertType(), mlir::remark::failed(), mlir::get(), mlir::LLVMTypeConverter::getMemRefAddressSpace(), and mlir::OpBuilder::setInsertionPointToStart().

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

getUniqueSymbolName()

static 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()

static 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::LLVMTypeConverter::convertType(), mlir::OpBuilder::create(), mlir::Operation::getAttrs(), mlir::OperationName::getIdentifier(), mlir::Builder::getIndexType(), mlir::Operation::getLoc(), mlir::Operation::getName(), and mlir::Operation::getResult().

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

wrapNumericMemorySpace()

static IntegerAttr wrapNumericMemorySpace ( MLIRContext *  ctx, unsigned  space  )

static

Definition at line 664 of file GPUOpsLowering.cpp.

References mlir::get().

Referenced by mlir::populateGpuMemorySpaceAttributeConversions().