Enable iterator space fusion for SD (huggingface#432)

qedawkins · web-flow · commit fbd77dc93631 · 2022-10-26T01:08:26.000-04:00
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -29,6 +29,7 @@ def get_models():`
`29`	`29`	`"--iree-flow-enable-conv-nchw-to-nhwc-transform",`
`30`	`30`	`"--iree-flow-enable-padding-linalg-ops",`
`31`	`31`	`"--iree-flow-linalg-ops-padding-size=16",`
	`32`	`+ "--iree-flow-enable-iterator-space-fusion",`
`32`	`33`	`]`
`33`	`34`	`if args.import_mlir == True:`
`34`	`35`	`return get_vae16(model_name=VAE_FP16), get_unet16_wrapped(`