[int8] Add option to generalize matmul ops

nod-ai · Jul 12, 2024 · 5ed406e · 5ed406e
1 parent 4c6c275
commit 5ed406e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 \