[mlir] Fix GpuToLLVM conversion pass after ControlFlow operations were split from Standard

This commit is contained in:
River Riddle 2022-02-06 15:10:03 -08:00
parent fe23a6fb75
commit 330838eb90
2 changed files with 3 additions and 0 deletions

View File

@ -31,6 +31,7 @@ add_mlir_conversion_library(MLIRGPUToGPURuntimeTransforms
LINK_LIBS PUBLIC
MLIRArithmeticToLLVM
MLIRAsyncToLLVM
MLIRControlFlowToLLVM
MLIRGPUTransforms
MLIRIR
MLIRLLVMCommonConversion

View File

@ -18,6 +18,7 @@
#include "../PassDetail.h"
#include "mlir/Conversion/ArithmeticToLLVM/ArithmeticToLLVM.h"
#include "mlir/Conversion/AsyncToLLVM/AsyncToLLVM.h"
#include "mlir/Conversion/ControlFlowToLLVM/ControlFlowToLLVM.h"
#include "mlir/Conversion/LLVMCommon/ConversionTarget.h"
#include "mlir/Conversion/LLVMCommon/Pattern.h"
#include "mlir/Conversion/MemRefToLLVM/MemRefToLLVM.h"
@ -351,6 +352,7 @@ void GpuToLLVMConversionPass::runOnOperation() {
target.addIllegalDialect<gpu::GPUDialect>();
mlir::arith::populateArithmeticToLLVMConversionPatterns(converter, patterns);
mlir::cf::populateControlFlowToLLVMConversionPatterns(converter, patterns);
populateVectorToLLVMConversionPatterns(converter, patterns);
populateMemRefToLLVMConversionPatterns(converter, patterns);
populateStdToLLVMConversionPatterns(converter, patterns);