[mlir][AMDGPU] Add AMDGPU conversion patterns to ConvertGPUToROCDL

This ensures that attributes such as the index bitwidth propagate
correctly to the AMDGPUToROCDL patterns.

Differential Revision: https://reviews.llvm.org/D125320
This commit is contained in:
Krzysztof Drewniak
2022-05-10 15:37:53 +00:00
parent 8200e1253f
commit 814b605095
3 changed files with 4 additions and 0 deletions

View File

@@ -11,6 +11,7 @@ add_mlir_conversion_library(MLIRGPUToROCDLTransforms
LINK_LIBS PUBLIC
MLIRArithmeticToLLVM
MLIRAMDGPUToROCDL
MLIRFuncToLLVM
MLIRGPUOps
MLIRGPUToGPURuntimeTransforms

View File

@@ -14,6 +14,7 @@
#include "mlir/Conversion/ControlFlowToLLVM/ControlFlowToLLVM.h"
#include "mlir/Conversion/GPUToROCDL/GPUToROCDLPass.h"
#include "mlir/Conversion/AMDGPUToROCDL/AMDGPUToROCDL.h"
#include "mlir/Conversion/ArithmeticToLLVM/ArithmeticToLLVM.h"
#include "mlir/Conversion/FuncToLLVM/ConvertFuncToLLVM.h"
#include "mlir/Conversion/LLVMCommon/ConversionTarget.h"
@@ -79,6 +80,7 @@ struct LowerGpuOpsToROCDLOpsPass
mlir::arith::populateArithmeticToLLVMConversionPatterns(converter,
llvmPatterns);
populateAMDGPUToROCDLConversionPatterns(converter, llvmPatterns);
populateVectorToLLVMConversionPatterns(converter, llvmPatterns);
populateVectorToROCDLConversionPatterns(converter, llvmPatterns);
cf::populateControlFlowToLLVMConversionPatterns(converter, llvmPatterns);

View File

@@ -3798,6 +3798,7 @@ cc_library(
includes = ["include"],
deps = [
":ArithmeticToLLVM",
":AMDGPUToROCDL",
":ControlFlowToLLVM",
":ConversionPassIncGen",
":FuncDialect",