#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"
Include dependency graph for GPUToSPIRV.cpp:
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__" |
Function Documentation
◆ getDefaultABIAttrs()
|
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().
◆ lowerAsEntryFunction()
|
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().
Variable Documentation
◆ kSPIRVModule
|
static |
Definition at line 25 of file GPUToSPIRV.cpp.
Referenced by getDefaultABIAttrs().
