Skip to content

Commit 161c7a1

Browse files
committed
add missing passes in nv pipeline
1 parent 5acc47a commit 161c7a1

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/jit/mlir.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -782,10 +782,15 @@ static const std::string gpu_pipeline =
782782
"func.func(insert-gpu-copy),"
783783
"drop-regions,"
784784
"canonicalize,"
785+
"func.func(lower-affine),"
785786
"gpu-kernel-outlining,"
787+
"canonicalize,"
788+
"cse,"
786789
"gpu.module(strip-debuginfo,convert-gpu-to-nvvm),"
787790
"nvvm-attach-target,"
788791
"func.func(gpu-async-region),"
792+
"expand-strided-metadata,"
793+
"lower-affine,"
789794
"gpu-to-llvm,"
790795
"gpu-module-to-binary{format=fatbin},"
791796
"convert-func-to-llvm,"

0 commit comments

Comments
 (0)