From 20025eed9f255c07ed72a3dcf8c7cbb05fb7c40b Mon Sep 17 00:00:00 2001
From: Emilien Bauer <bauer.emilien@gmail.com>
Date: Fri, 30 Aug 2024 14:47:02 +0100
Subject: [PATCH] GPU pipeline canonicalization

---
 devito/xdsl_core/xdsl_gpu.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/devito/xdsl_core/xdsl_gpu.py b/devito/xdsl_core/xdsl_gpu.py
index 9f281015b3..5cd7b61245 100644
--- a/devito/xdsl_core/xdsl_gpu.py
+++ b/devito/xdsl_core/xdsl_gpu.py
@@ -140,6 +140,8 @@ def _jit_compile(self):
 
 def generate_XDSL_GPU_PIPELINE():
     passes = [
+        "canonicalize",
+        "cse",
         "shape-inference",
         "convert-stencil-to-ll-mlir",
         "reconcile-unrealized-casts",