add rest

nod-ai · Aug 2, 2024 · 94a0a9a · 94a0a9a
1 parent 94caa22
commit 94a0a9a
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/iree_special_models/sd3/mmdit/test_mmdit.py b/iree_special_models/sd3/mmdit/test_mmdit.py
@@ -52,7 +52,7 @@
     "--iree-codegen-llvmgpu-use-vector-distribution",
     "--iree-rocm-waves-per-eu=2",
     "--iree-execution-model=async-external",
-    "--iree-preprocessing-pass-pipeline=\"builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics)\"",
+    "--iree-preprocessing-pass-pipeline='builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics)'",
 ]
 
 mlir_path = current_dir + "/model.mlirbc"

diff --git a/iree_special_models/sd3/prompt-encoder/test_clip.py b/iree_special_models/sd3/prompt-encoder/test_clip.py
@@ -51,7 +51,7 @@
     "--iree-opt-aggressively-propagate-transposes=true",
     "--iree-codegen-llvmgpu-use-vector-distribution=true",
     "--iree-execution-model=async-external",
-    "--iree-preprocessing-pass-pipeline=\"builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics{pad-target-type=conv})\"",
+    "--iree-preprocessing-pass-pipeline='builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics{pad-target-type=conv})'",
 ]
 
 mlir_path = current_dir + "/model.mlirbc"

diff --git a/iree_special_models/sd3/vae-decode/test_vae.py b/iree_special_models/sd3/vae-decode/test_vae.py
@@ -42,7 +42,7 @@
     "--iree-flow-enable-aggressive-fusion=true",
     "--iree-codegen-llvmgpu-use-vector-distribution=true",
     "--iree-execution-model=async-external",
-    "--iree-preprocessing-pass-pipeline=\"builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics)\"",
+    "--iree-preprocessing-pass-pipeline='builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics)'",
 ]
 
 mlir_path = current_dir + "/model.mlirbc"

diff --git a/iree_special_models/sdxl/prompt-encoder/test_clip.py b/iree_special_models/sdxl/prompt-encoder/test_clip.py
@@ -49,7 +49,7 @@
     "--iree-opt-aggressively-propagate-transposes=true",
     "--iree-codegen-llvmgpu-use-vector-distribution=true",
     "--iree-execution-model=async-external",
-    "--iree-preprocessing-pass-pipeline=\"builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics{pad-target-type=conv})\"",
+    "--iree-preprocessing-pass-pipeline='builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics{pad-target-type=conv})'",
     "--iree-scheduling-dump-statistics-format=json",
     "--iree-scheduling-dump-statistics-file=compilation_info.json"
 ]

diff --git a/iree_special_models/sdxl/scheduled-unet/test_unet.py b/iree_special_models/sdxl/scheduled-unet/test_unet.py
@@ -53,7 +53,7 @@
     "--iree-codegen-llvmgpu-use-vector-distribution",
     "--iree-rocm-waves-per-eu=2",
     "--iree-execution-model=async-external",
-    "--iree-preprocessing-pass-pipeline=\"builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics)\"",
+    "--iree-preprocessing-pass-pipeline='builtin.module(iree-preprocessing-transpose-convolution-pipeline,iree-preprocessing-pad-to-intrinsics)'",
     "--iree-scheduling-dump-statistics-format=json",
     "--iree-scheduling-dump-statistics-file=compilation_info.json"
 ]