Enable iterator space fusion for SD (#432)

2026-04-03 03:00:17 -04:00 · 2022-10-26 01:08:26 -04:00
parent cdc6dd19e3
commit fbd77dc936
1 changed files with 1 additions and 0 deletions
--- a/shark/examples/shark_inference/stable_diffusion/main.py
+++ b/shark/examples/shark_inference/stable_diffusion/main.py
@@ -29,6 +29,7 @@ def get_models():
            "--iree-flow-enable-conv-nchw-to-nhwc-transform",
            "--iree-flow-enable-padding-linalg-ops",
            "--iree-flow-linalg-ops-padding-size=16",
+            "--iree-flow-enable-iterator-space-fusion",
        ]
        if args.import_mlir == True:
            return get_vae16(model_name=VAE_FP16), get_unet16_wrapped(