#include "mlir/Conversion/GPUToSPIRV/GPUToSPIRV.h"#include "mlir/Dialect/GPU/IR/GPUDialect.h"#include "mlir/Dialect/SPIRV/IR/SPIRVAttributes.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/IR/Matchers.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 spirv.interface_var_abi attributes for lowering gpu.func to spirv.func if no arguments have the attributes set already. More... | |
Variables | |
| static constexpr const char | kSPIRVModule [] = "__spv__" |
Function Documentation
◆ getDefaultABIAttrs()
|
static |
Populates argABI with spirv.interface_var_abi attributes for lowering gpu.func to spirv.func if no arguments have the attributes set already.
Returns failure if any argument has the ABI attribute set already.
Definition at line 256 of file GPUToSPIRV.cpp.
References mlir::failure(), mlir::spirv::getInterfaceVarABIAttr(), mlir::spirv::getInterfaceVarABIAttrName(), mlir::spirv::lookupTargetEnvOrDefault(), mlir::spirv::needsInterfaceVarABIAttrs(), and mlir::success().
◆ lowerAsEntryFunction()
|
static |
Definition at line 195 of file GPUToSPIRV.cpp.
References mlir::TypeConverter::SignatureConversion::addInputs(), mlir::ConversionPatternRewriter::convertRegionTypes(), mlir::TypeConverter::convertType(), mlir::OpBuilder::create(), mlir::detail::enumerate(), mlir::ConversionPatternRewriter::eraseOp(), mlir::failed(), mlir::TypeConverter::SignatureConversion::getConvertedTypes(), mlir::spirv::getEntryPointABIAttrName(), mlir::Builder::getFunctionType(), mlir::spirv::getInterfaceVarABIAttrName(), mlir::SymbolTable::getSymbolAttrName(), mlir::function_interface_impl::getTypeAttrName(), mlir::ConversionPatternRewriter::inlineRegionBefore(), and mlir::Operation::setAttr().
Variable Documentation
◆ kSPIRVModule
|
staticconstexpr |
Definition at line 27 of file GPUToSPIRV.cpp.
