diff --git a/int8-model/compile-punet-base.sh b/int8-model/compile-punet-base.sh
index b41c7d3..f98533d 100755
--- a/int8-model/compile-punet-base.sh
+++ b/int8-model/compile-punet-base.sh
@@ -44,6 +44,7 @@ set -x
     --iree-rocm-bc-dir="${SCRIPT_DIR}/../bitcode-6.1.2" \
     --iree-opt-const-eval=false \
     --iree-opt-data-tiling=false \
+    --iree-global-opt-enable-generalize-linalg-matmul-ops=true \
     --iree-global-opt-propagate-transposes=true \
     --iree-opt-aggressively-propagate-transposes=true \
     --iree-flow-enable-aggressive-fusion \