Add CSE and canonicalize before final conversion to binary.

mshahneo · mshahneo · commit 411ded5110a1 · 2025-10-29T20:27:44.000Z
diff --git a/mlir/lib/Dialect/GPU/Pipelines/GPUToXeVMPipeline.cpp b/mlir/lib/Dialect/GPU/Pipelines/GPUToXeVMPipeline.cpp
@@ -106,6 +106,8 @@ void buildPostGPUCommonPassPipeline(
   pm.addPass(createLowerAffinePass());
   pm.addPass(createConvertToLLVMPass());
   pm.addPass(createReconcileUnrealizedCastsPass());
+  pm.addNestedPass<gpu::GPUModuleOp>(createCanonicalizerPass());
+  pm.addNestedPass<gpu::GPUModuleOp>(createCSEPass());
   // gpu-module-to-binary
   {
     GpuModuleToBinaryPassOptions gpuToModuleBinOptions;

Original file line number	Diff line number	Diff line change
`@@ -106,6 +106,8 @@ void buildPostGPUCommonPassPipeline(`
`106`	`106`	`pm.addPass(createLowerAffinePass());`
`107`	`107`	`pm.addPass(createConvertToLLVMPass());`
`108`	`108`	`pm.addPass(createReconcileUnrealizedCastsPass());`
	`109`	`+ pm.addNestedPass<gpu::GPUModuleOp>(createCanonicalizerPass());`
	`110`	`+ pm.addNestedPass<gpu::GPUModuleOp>(createCSEPass());`
`109`	`111`	`// gpu-module-to-binary`
`110`	`112`	`{`
`111`	`113`	`GpuModuleToBinaryPassOptions gpuToModuleBinOptions;`