|
12 | 12 | //===----------------------------------------------------------------------===// |
13 | 13 |
|
14 | 14 | #include "mlir/Conversion/GPUToSPIRV/GPUToSPIRVPass.h" |
15 | | - |
16 | 15 | #include "mlir/Conversion/ArithToSPIRV/ArithToSPIRV.h" |
| 16 | +#include "mlir/Conversion/ComplexToSPIRV/ComplexToSPIRV.h" |
| 17 | +#include "mlir/Conversion/ControlFlowToSPIRV/ControlFlowToSPIRV.h" |
17 | 18 | #include "mlir/Conversion/FuncToSPIRV/FuncToSPIRV.h" |
18 | 19 | #include "mlir/Conversion/GPUToSPIRV/GPUToSPIRV.h" |
| 20 | +#include "mlir/Conversion/IndexToSPIRV/IndexToSPIRV.h" |
| 21 | +#include "mlir/Conversion/MathToSPIRV/MathToSPIRV.h" |
19 | 22 | #include "mlir/Conversion/MemRefToSPIRV/MemRefToSPIRV.h" |
20 | 23 | #include "mlir/Conversion/SCFToSPIRV/SCFToSPIRV.h" |
| 24 | +#include "mlir/Conversion/TensorToSPIRV/TensorToSPIRV.h" |
| 25 | +#include "mlir/Conversion/UBToSPIRV/UBToSPIRV.h" |
21 | 26 | #include "mlir/Conversion/VectorToSPIRV/VectorToSPIRV.h" |
22 | 27 | #include "mlir/Dialect/Func/IR/FuncOps.h" |
23 | 28 | #include "mlir/Dialect/GPU/IR/GPUDialect.h" |
@@ -159,6 +164,15 @@ void GPUToSPIRVPass::runOnOperation() { |
159 | 164 | populateFuncToSPIRVPatterns(typeConverter, patterns); |
160 | 165 | populateVectorToSPIRVPatterns(typeConverter, patterns); |
161 | 166 |
|
| 167 | + mlir::populateBuiltinFuncToSPIRVPatterns(typeConverter, patterns); |
| 168 | + mlir::populateComplexToSPIRVPatterns(typeConverter, patterns); |
| 169 | + mlir::cf::populateControlFlowToSPIRVPatterns(typeConverter, patterns); |
| 170 | + mlir::index::populateIndexToSPIRVPatterns(typeConverter, patterns); |
| 171 | + mlir::populateMathToSPIRVPatterns(typeConverter, patterns); |
| 172 | + mlir::populateTensorToSPIRVPatterns(typeConverter, |
| 173 | + /*byteCountThreshold=*/64, patterns); |
| 174 | + mlir::ub::populateUBToSPIRVConversionPatterns(typeConverter, patterns); |
| 175 | + |
162 | 176 | if (failed(applyFullConversion(gpuModule, *target, std::move(patterns)))) |
163 | 177 | return signalPassFailure(); |
164 | 178 | } |
|
0 commit comments