From 1a8e2024d6e4205c32c8ce956be7c6147cf063ee Mon Sep 17 00:00:00 2001
From: Ean Garvey <87458719+monorimet@users.noreply.github.com>
Date: Wed, 21 Jun 2023 11:36:55 -0500
Subject: [PATCH] Exclude non-square sizes from use_tuned on rdna2 (#1568)

---
 apps/stable_diffusion/src/utils/utils.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/apps/stable_diffusion/src/utils/utils.py b/apps/stable_diffusion/src/utils/utils.py
index 85d943d93c..aca9169a9c 100644
--- a/apps/stable_diffusion/src/utils/utils.py
+++ b/apps/stable_diffusion/src/utils/utils.py
@@ -305,6 +305,11 @@ def set_init_device_flags():
     ):
         args.use_tuned = False
 
+    elif (
+        args.height != args.width and "rdna2" in args.iree_vulkan_target_triple
+    ):
+        args.use_tuned = False
+
     elif base_model_id not in [
         "Linaqruf/anything-v3.0",
         "dreamlike-art/dreamlike-diffusion-1.0",