GPU pipeline canonicalization

xdslproject · Aug 30, 2024 · 51e6f97 · 51e6f97
1 parent 89e774c
commit 51e6f97
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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",