#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 |
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().
