MLIR
16.0.0git
|
#include "mlir/Conversion/GPUToSPIRV/GPUToSPIRV.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVEnums.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
#include "mlir/Dialect/SPIRV/IR/TargetAndABI.h"
#include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/Transforms/DialectConversion.h"
Go to the source code of this file.
Functions | |
static spirv::FuncOp | lowerAsEntryFunction (gpu::GPUFuncOp funcOp, TypeConverter &typeConverter, ConversionPatternRewriter &rewriter, spirv::EntryPointABIAttr entryPointInfo, ArrayRef< spirv::InterfaceVarABIAttr > argABIInfo) |
static LogicalResult | getDefaultABIAttrs (MLIRContext *context, gpu::GPUFuncOp funcOp, SmallVectorImpl< spirv::InterfaceVarABIAttr > &argABI) |
Populates argABI with spv.interface_var_abi attributes for lowering gpu.func to spv.func if no arguments have the attributes set already. More... | |
Variables | |
static constexpr const char | kSPIRVModule [] = "__spv__" |
|
static |
Populates argABI
with spv.interface_var_abi attributes for lowering gpu.func to spv.func if no arguments have the attributes set already.
Returns failure if any argument has the ABI attribute set already.
Definition at line 244 of file GPUToSPIRV.cpp.
References mlir::OpBuilder::create(), mlir::ConversionPatternRewriter::eraseBlock(), mlir::ConversionPatternRewriter::eraseOp(), mlir::failed(), mlir::failure(), mlir::spirv::getAddressingModel(), mlir::Builder::getContext(), mlir::spirv::getInterfaceVarABIAttr(), mlir::spirv::getInterfaceVarABIAttrName(), mlir::spirv::getMemoryModel(), mlir::Builder::getStringAttr(), mlir::ConversionPatternRewriter::inlineRegionBefore(), kSPIRVModule, mlir::spirv::lookupEntryPointABI(), mlir::spirv::lookupTargetEnvOrDefault(), lowerAsEntryFunction(), mlir::spirv::needsInterfaceVarABIAttrs(), mlir::RewriterBase::replaceOpWithNewOp(), and mlir::success().
|
static |
Definition at line 183 of file GPUToSPIRV.cpp.
References mlir::ConversionPatternRewriter::convertRegionTypes(), mlir::TypeConverter::convertType(), mlir::OpBuilder::create(), mlir::detail::enumerate(), mlir::ConversionPatternRewriter::eraseOp(), mlir::failed(), mlir::spirv::getEntryPointABIAttrName(), mlir::Builder::getFunctionType(), mlir::spirv::getInterfaceVarABIAttrName(), mlir::SymbolTable::getSymbolAttrName(), mlir::function_interface_impl::getTypeAttrName(), and mlir::ConversionPatternRewriter::inlineRegionBefore().
Referenced by getDefaultABIAttrs().
|
static |
Definition at line 25 of file GPUToSPIRV.cpp.
Referenced by getDefaultABIAttrs().