MLIR  17.0.0git
Functions
WmmaOpsToSPIRV.cpp File Reference
#include "mlir/Conversion/GPUToSPIRV/GPUToSPIRV.h"
#include "mlir/Conversion/GPUToSPIRV/GPUToSPIRVPass.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/TypeUtilities.h"
+ Include dependency graph for WmmaOpsToSPIRV.cpp:

Go to the source code of this file.

Functions

static bool createElementwiseOp (ConversionPatternRewriter &builder, gpu::SubgroupMmaElementwiseOp op, spirv::CooperativeMatrixNVType coopType, ValueRange operands)
 Creates a SPIR-V op to replace the given GPU subgroup mma elementwise op when the elementwise op directly supports with cooperative matrix type. More...
 

Function Documentation

◆ createElementwiseOp()

static bool createElementwiseOp ( ConversionPatternRewriter builder,
gpu::SubgroupMmaElementwiseOp  op,
spirv::CooperativeMatrixNVType  coopType,
ValueRange  operands 
)
static

Creates a SPIR-V op to replace the given GPU subgroup mma elementwise op when the elementwise op directly supports with cooperative matrix type.

Returns false if cannot.

See SPV_NV_cooperative_matrix for supported elementwise ops.

Definition at line 32 of file WmmaOpsToSPIRV.cpp.

References mlir::RewriterBase::replaceOpWithNewOp().