Skip to content

Commit 60a28d4

Browse files
committed
revert change to horizontal contraction fusion flag pending flag change in IREE main.
1 parent 9fdc07f commit 60a28d4

File tree

1 file changed

+3
-3
lines changed
  • models/turbine_models/custom_models/sd_inference

1 file changed

+3
-3
lines changed

models/turbine_models/custom_models/sd_inference/utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,13 +36,13 @@
3636
],
3737
"unet": [
3838
"--iree-flow-enable-aggressive-fusion",
39-
"--iree-flow-enable-fuse-horizontal-contractions=true",
39+
"--iree-global-opt-enable-fuse-horizontal-contractions=true",
4040
"--iree-opt-aggressively-propagate-transposes=true",
4141
"--iree-codegen-llvmgpu-use-vector-distribution=true",
4242
],
4343
"clip": [
4444
"--iree-flow-enable-aggressive-fusion",
45-
"--iree-flow-enable-fuse-horizontal-contractions=true",
45+
"--iree-global-opt-enable-fuse-horizontal-contractions=true",
4646
"--iree-opt-aggressively-propagate-transposes=true",
4747
],
4848
"vae": [
@@ -61,7 +61,7 @@
6161
"--iree-opt-const-eval=false",
6262
"--iree-opt-aggressively-propagate-transposes=true",
6363
"--iree-flow-enable-aggressive-fusion",
64-
"--iree-flow-enable-fuse-horizontal-contractions=true",
64+
"--iree-global-opt-enable-fuse-horizontal-contractions=true",
6565
"--iree-codegen-gpu-native-math-precision=true",
6666
"--iree-codegen-llvmgpu-use-vector-distribution=true",
6767
"--iree-codegen-llvmgpu-enable-transform-dialect-jit=false",

0 commit comments

Comments
 (0)