Update unet horizontal fusion flag (#790)

nod-ai · Jul 25, 2024 · 920dbf5 · 920dbf5
1 parent 0e57b4e
commit 920dbf5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py
@@ -40,6 +40,7 @@
         "--iree-opt-data-tiling=false",
         "--iree-codegen-gpu-native-math-precision=true",
         "--iree-vm-target-truncate-unsupported-floats",
+        "--iree-flow-enable-fuse-horizontal-contractions=true",
     ],
     "clip": [
         "--iree-flow-enable-aggressive-fusion",