From 51e6f97ded417939105087eabc5a7fe63f730412 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 | 3 +++
 1 file changed, 3 insertions(+)

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