use one-shot-bufferize pass

Author: tkarna

src/jit/mlir.cpp

@@ -687,19 +687,8 @@ static const std::string cpu_pipeline =
     // "convert-shape-to-std,"
     "arith-expand,"
     "memref-expand,"
-    "arith-bufferize,"
-    // "func-bufferize,"
     "func.func(empty-tensor-to-alloc-tensor),"
-    "func.func(scf-bufferize),"
-    "func.func(tensor-bufferize),"
-    "func.func(bufferization-bufferize),"
-    "func.func(linalg-bufferize),"
-    "func.func(linalg-detensorize),"
-    "func.func(tensor-bufferize),"
-    "region-bufferize,"
-    "canonicalize,"
-    "func.func(finalizing-bufferize),"
-    "func.func(buffer-deallocation),"
+    "one-shot-bufferize{bufferize-function-boundaries},"
     "imex-remove-temporaries,"
     "func.func(convert-linalg-to-parallel-loops),"
     "func.func(scf-parallel-loop-fusion),"