GPUToLLVMConversion.cpp File Reference

#include "mlir/Conversion/GPUCommon/GPUCommonPass.h"
#include "mlir/Conversion/ArithToLLVM/ArithToLLVM.h"
#include "mlir/Conversion/AsyncToLLVM/AsyncToLLVM.h"
#include "mlir/Conversion/ControlFlowToLLVM/ControlFlowToLLVM.h"
#include "mlir/Conversion/FuncToLLVM/ConvertFuncToLLVM.h"
#include "mlir/Conversion/FuncToLLVM/ConvertFuncToLLVMPass.h"
#include "mlir/Conversion/LLVMCommon/ConversionTarget.h"
#include "mlir/Conversion/LLVMCommon/Pattern.h"
#include "mlir/Conversion/MemRefToLLVM/MemRefToLLVM.h"
#include "mlir/Conversion/VectorToLLVM/ConvertVectorToLLVM.h"
#include "mlir/Dialect/Async/IR/Async.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/GPU/Transforms/Passes.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/IR/Attributes.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/IR/BuiltinTypes.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/Support/Error.h"
#include "llvm/Support/FormatVariadic.h"
#include "mlir/Conversion/Passes.h.inc"

Go to the source code of this file.

Namespaces
	mlir
	This header declares functions that assist transformations in the MemRef dialect.

Macros
#define	GEN_PASS_DEF_GPUTOLLVMCONVERSIONPASS

Functions
static int32_t	getCuSparseIndexTypeFrom (Type type)
static int32_t	getCuSparseLtDataTypeFrom (Type type)
static int32_t	getCuSparseDataTypeFrom (Type type)
static bool	is2To4Sparsity (Value spMat)
static bool	isSpMMCusparseLtOp (Value op)
static LogicalResult	areAllLLVMTypes (Operation *op, ValueRange operands, ConversionPatternRewriter &rewriter)
static LogicalResult	isAsyncWithOneDependency (ConversionPatternRewriter &rewriter, gpu::AsyncOpInterface op)
static bool	isGpuAsyncTokenType (Value value)
static bool	isDefinedByCallTo (Value value, StringRef functionName)
static Value	bitAndAddrspaceCast (Location loc, ConversionPatternRewriter &rewriter, LLVM::LLVMPointerType destinationType, Value sourcePtr, LLVMTypeConverter &typeConverter)
template<typename T >
static Value	genConstInt32From (OpBuilder &builder, Location loc, T TValue)
static Value	genConstInt32FromComputeMode (OpBuilder &builder, Location loc, Type computeType)
template<typename T >
static void	addOpaquePointerConversion (LLVMTypeConverter &converter)

Variables
static constexpr const char *	kGpuBinaryStorageSuffix = "_gpubin_cst"

Macro Definition Documentation

GEN_PASS_DEF_GPUTOLLVMCONVERSIONPASS

#define GEN_PASS_DEF_GPUTOLLVMCONVERSIONPASS

Definition at line 42 of file GPUToLLVMConversion.cpp.

Function Documentation

addOpaquePointerConversion()

template<typename T >

static void addOpaquePointerConversion ( LLVMTypeConverter &  converter )

static

Definition at line 1881 of file GPUToLLVMConversion.cpp.

References mlir::TypeConverter::addConversion(), mlir::get(), mlir::LLVMTypeConverter::getContext(), and mlir::LLVMTypeConverter::getPointerType().

areAllLLVMTypes()

static LogicalResult areAllLLVMTypes ( Operation *  op, ValueRange  operands, ConversionPatternRewriter &  rewriter  )

static

Definition at line 868 of file GPUToLLVMConversion.cpp.

bitAndAddrspaceCast()

static Value bitAndAddrspaceCast ( Location  loc, ConversionPatternRewriter &  rewriter, LLVM::LLVMPointerType  destinationType, Value  sourcePtr, LLVMTypeConverter &  typeConverter  )

static

Definition at line 1304 of file GPUToLLVMConversion.cpp.

References mlir::OpBuilder::create(), mlir::LLVMTypeConverter::getPointerType(), mlir::Value::getType(), and mlir::LLVMTypeConverter::useOpaquePointers().

genConstInt32From()

template<typename T >

static Value genConstInt32From ( OpBuilder &  builder, Location  loc, T  TValue  )

static

Definition at line 1414 of file GPUToLLVMConversion.cpp.

References mlir::OpBuilder::create(), and mlir::Builder::getIntegerType().

Referenced by genConstInt32FromComputeMode().

genConstInt32FromComputeMode()

static Value genConstInt32FromComputeMode ( OpBuilder &  builder, Location  loc, Type  computeType  )

static

Definition at line 1420 of file GPUToLLVMConversion.cpp.

References genConstInt32From(), and getCuSparseDataTypeFrom().

getCuSparseDataTypeFrom()

static int32_t getCuSparseDataTypeFrom ( Type  type )

static

Definition at line 797 of file GPUToLLVMConversion.cpp.

References mlir::Type::cast(), getElementType(), mlir::Type::isBF16(), mlir::Type::isF16(), mlir::Type::isF32(), mlir::Type::isF64(), and mlir::Type::isInteger().

Referenced by genConstInt32FromComputeMode().

getCuSparseIndexTypeFrom()

static int32_t getCuSparseIndexTypeFrom ( Type  type )

static

Definition at line 779 of file GPUToLLVMConversion.cpp.

References mlir::Type::isa().

getCuSparseLtDataTypeFrom()

static int32_t getCuSparseLtDataTypeFrom ( Type  type )

static

Definition at line 787 of file GPUToLLVMConversion.cpp.

References mlir::Type::isF16(), and mlir::Type::isInteger().

is2To4Sparsity()

static bool is2To4Sparsity ( Value  spMat )

static

Definition at line 841 of file GPUToLLVMConversion.cpp.

isAsyncWithOneDependency()

static LogicalResult isAsyncWithOneDependency ( ConversionPatternRewriter &  rewriter, gpu::AsyncOpInterface  op  )

static

Definition at line 879 of file GPUToLLVMConversion.cpp.

isDefinedByCallTo()

static bool isDefinedByCallTo ( Value  value, StringRef  functionName  )

static

Definition at line 1037 of file GPUToLLVMConversion.cpp.

References mlir::Value::getDefiningOp(), and mlir::Value::getType().

isGpuAsyncTokenType()

static bool isGpuAsyncTokenType ( Value  value )

static

Definition at line 1001 of file GPUToLLVMConversion.cpp.

References mlir::Value::getType().

isSpMMCusparseLtOp()

static bool isSpMMCusparseLtOp ( Value  op )

static

Definition at line 855 of file GPUToLLVMConversion.cpp.

Variable Documentation

kGpuBinaryStorageSuffix

constexpr const char* kGpuBinaryStorageSuffix = "_gpubin_cst"

staticconstexpr

Definition at line 48 of file GPUToLLVMConversion.cpp.