gpu pipeline uses one-shot-bufferize

src/jit/mlir.cpp

@@ -725,18 +725,8 @@ static const std::string gpu_pipeline =
     "linalg-fuse-elementwise-ops,"
     "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),"
+    "one-shot-bufferize{bufferize-function-boundaries},"
     "imex-remove-temporaries,"
     "func.func(convert-linalg-to-parallel-loops),"
     "func.func(scf-parallel-loop-fusion),"