MLIR
18.0.0git
|
#include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h"
#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
#include "mlir/Transforms/DialectConversion.h"
Go to the source code of this file.
Namespaces | |
mlir | |
Include the generated interface declarations. | |
Functions | |
void | mlir::populateGPUToSPIRVPatterns (SPIRVTypeConverter &typeConverter, RewritePatternSet &patterns) |
Appends to a pattern list additional patterns for translating GPU Ops to SPIR-V ops. More... | |
void | mlir::populateGpuWMMAToSPIRVCoopMatrixKHRConversionPatterns (SPIRVTypeConverter &typeConverter, RewritePatternSet &patterns) |
Collect a set of patterns to convert WMMA ops from GPU dialect to SPIRV, using the KHR Cooperative Matrix extension. More... | |
void | mlir::populateGpuWMMAToSPIRVCoopMatrixNVConversionPatterns (SPIRVTypeConverter &typeConverter, RewritePatternSet &patterns) |
Collect a set of patterns to convert WMMA ops from GPU dialect to SPIRV, using the NV Cooperative Matrix extension. More... | |
void | mlir::populateMMAToSPIRVCoopMatrixTypeConversion (SPIRVTypeConverter &typeConverter, bool useNVTypes=false) |
Adds MMAMatrixType conversions to SPIR-V cooperative matrix type conversion to the type converter. More... | |