From 60a28d4868e73cf585d3e0875195f6206cfeab79 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Wed, 17 Jul 2024 12:22:06 -0500 Subject: [PATCH] revert change to horizontal contraction fusion flag pending flag change in IREE main. --- models/turbine_models/custom_models/sd_inference/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py index 448b6791..78f20a12 100644 --- a/models/turbine_models/custom_models/sd_inference/utils.py +++ b/models/turbine_models/custom_models/sd_inference/utils.py @@ -36,13 +36,13 @@ ], "unet": [ "--iree-flow-enable-aggressive-fusion", - "--iree-flow-enable-fuse-horizontal-contractions=true", + "--iree-global-opt-enable-fuse-horizontal-contractions=true", "--iree-opt-aggressively-propagate-transposes=true", "--iree-codegen-llvmgpu-use-vector-distribution=true", ], "clip": [ "--iree-flow-enable-aggressive-fusion", - "--iree-flow-enable-fuse-horizontal-contractions=true", + "--iree-global-opt-enable-fuse-horizontal-contractions=true", "--iree-opt-aggressively-propagate-transposes=true", ], "vae": [ @@ -61,7 +61,7 @@ "--iree-opt-const-eval=false", "--iree-opt-aggressively-propagate-transposes=true", "--iree-flow-enable-aggressive-fusion", - "--iree-flow-enable-fuse-horizontal-contractions=true", + "--iree-global-opt-enable-fuse-horizontal-contractions=true", "--iree-codegen-gpu-native-math-precision=true", "--iree-codegen-llvmgpu-use-vector-distribution=true", "--iree-codegen-llvmgpu-enable-transform-dialect-jit=false",