Skip to content

Commit d33deea

Browse files
committed
add missing passes in nv pipeline
1 parent e9eac9a commit d33deea

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/jit/mlir.cpp

+5
Original file line numberDiff line numberDiff line change
@@ -781,10 +781,15 @@ static const std::string gpu_pipeline =
781781
"func.func(insert-gpu-copy),"
782782
"drop-regions,"
783783
"canonicalize,"
784+
"func.func(lower-affine),"
784785
"gpu-kernel-outlining,"
786+
"canonicalize,"
787+
"cse,"
785788
"gpu.module(strip-debuginfo,convert-gpu-to-nvvm),"
786789
"nvvm-attach-target,"
787790
"func.func(gpu-async-region),"
791+
"expand-strided-metadata,"
792+
"lower-affine,"
788793
"gpu-to-llvm,"
789794
"gpu-module-to-binary{format=fatbin},"
790795
"convert-func-to-llvm,"

0 commit comments

Comments
 (0)