From 920dbf5d53057ed04d08c515f2bd9db70559fcd6 Mon Sep 17 00:00:00 2001
From: saienduri <77521230+saienduri@users.noreply.github.com>
Date: Wed, 24 Jul 2024 17:07:37 -0700
Subject: [PATCH] Update unet horizontal fusion flag (#790)

---
 models/turbine_models/custom_models/sd_inference/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py
index 9d5c149aa..2b6a164c1 100644
--- 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",