Add remove temporaries pass (#27)

Author: tkarna

CMakeLists.txt

@@ -174,6 +174,7 @@ target_link_libraries(_ddpt_rt PRIVATE
     IMEXDistDialect
     IMEXDistTransforms
     IMEXDistToStandard
+    IMEXTransforms
 	  MLIROptLib
     MLIRExecutionEngine
     MLIRIR

examples/wave_equation.py

@@ -202,6 +202,10 @@ def step(u, v, e, u1, v1, e1, u2, v2, e2):
         v[:, 1:ny] = v[:, 1:ny] / 3.0 + 2.0 / 3.0 * (v2[:, 1:ny] + dt * dvdt)
         e[:, :] = e[:, :] / 3.0 + 2.0 / 3.0 * (e2[:, :] + dt * dedt)
 
+        del dudt
+        del dvdt
+        del dedt
+
     t = 0
     i_export = 0
     next_t_export = 0

imex_version.txt

@@ -1 +1 @@
-c86b38a055b73c9041ca38dd36760fa11d3d5b9f
+999b959287d098c0f5062dff294e4c3453ba3f38

src/jit/mlir.cpp

@@ -427,6 +427,7 @@ static const char *pass_pipeline =
                             "func.func(tensor-bufferize),"
                             "func.func(finalizing-bufferize),"
                             "func.func(buffer-deallocation),"
+                            "imex-remove-temporaries,"
                             "func.func(convert-linalg-to-parallel-loops),"
                             "func.func(scf-parallel-loop-fusion),"
                             "canonicalize,"
@@ -528,6 +529,8 @@ void init() {
   ::imex::registerConvertDistToStandard();
   ::imex::registerConvertPTensorToLinalg();
 
+  ::imex::registerRemoveTemporariesPass();
+
   // ::mlir::DialectRegistry registry;
   // ::mlir::registerAllDialects(registry);
   // ::imex::registerAllDialects(registry);