[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:
@@ -11,6 +11,7 @@ add_mlir_conversion_library(MLIRGPUToROCDLTransforms
|
||||
|
||||
LINK_LIBS PUBLIC
|
||||
MLIRArithmeticToLLVM
|
||||
MLIRAMDGPUToROCDL
|
||||
MLIRFuncToLLVM
|
||||
MLIRGPUOps
|
||||
MLIRGPUToGPURuntimeTransforms
|
||||
|
||||
@@ -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);
|
||||
|
||||
@@ -3798,6 +3798,7 @@ cc_library(
|
||||
includes = ["include"],
|
||||
deps = [
|
||||
":ArithmeticToLLVM",
|
||||
":AMDGPUToROCDL",
|
||||
":ControlFlowToLLVM",
|
||||
":ConversionPassIncGen",
|
||||
":FuncDialect",
|
||||
|
||||
Reference in New Issue
Block a user