Skip to content

Commit 5483431

Browse files
committed
gpu pipeline uses one-shot-bufferize
1 parent d4428cc commit 5483431

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

src/jit/mlir.cpp

+1-11
Original file line numberDiff line numberDiff line change
@@ -725,18 +725,8 @@ static const std::string gpu_pipeline =
725725
"linalg-fuse-elementwise-ops,"
726726
"arith-expand,"
727727
"memref-expand,"
728-
"arith-bufferize,"
729-
"func-bufferize,"
730728
"func.func(empty-tensor-to-alloc-tensor),"
731-
"func.func(scf-bufferize),"
732-
"func.func(tensor-bufferize),"
733-
"func.func(bufferization-bufferize),"
734-
"func.func(linalg-bufferize),"
735-
"func.func(linalg-detensorize),"
736-
"func.func(tensor-bufferize),"
737-
"region-bufferize,"
738-
"canonicalize,"
739-
"func.func(finalizing-bufferize),"
729+
"one-shot-bufferize{bufferize-function-boundaries},"
740730
"imex-remove-temporaries,"
741731
"func.func(convert-linalg-to-parallel-loops),"
742732
"func.func(scf-parallel-loop-fusion),"

0 commit comments

Comments
 (0)