diff --git a/mlir/lib/Dialect/GPU/Pipelines/GPUToXeVMPipeline.cpp b/mlir/lib/Dialect/GPU/Pipelines/GPUToXeVMPipeline.cpp index 1a1485ba2e02c..1c46fa518dd99 100644 --- a/mlir/lib/Dialect/GPU/Pipelines/GPUToXeVMPipeline.cpp +++ b/mlir/lib/Dialect/GPU/Pipelines/GPUToXeVMPipeline.cpp @@ -85,6 +85,7 @@ void buildGPUPassPipeline(OpPassManager &pm, pm.addNestedPass( createConvertGpuOpsToLLVMSPVOps(gpuToLLVMSPVOptions)); } + pm.addNestedPass(createConvertToLLVMPass()); pm.addNestedPass(createCSEPass()); pm.addNestedPass(createReconcileUnrealizedCastsPass()); } @@ -106,6 +107,8 @@ void buildPostGPUCommonPassPipeline( pm.addPass(createLowerAffinePass()); pm.addPass(createConvertToLLVMPass()); pm.addPass(createReconcileUnrealizedCastsPass()); + pm.addNestedPass(createCanonicalizerPass()); + pm.addNestedPass(createCSEPass()); // gpu-module-to-binary { GpuModuleToBinaryPassOptions gpuToModuleBinOptions;