add flag for generalizing batch matmul ops (#52)

iree-org/iree#17877 --------- Co-authored-by: Quinn Dawkins <[email protected]>
nod-ai · Jul 12, 2024 · 7d15e77 · 7d15e77
1 parent 5ed406e
commit 7d15e77
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/int8-model/compile-punet-base.sh b/int8-model/compile-punet-base.sh
@@ -32,7 +32,7 @@ fi
 shift 4
 
 readonly DEFAULT_FLAGS=(
-"--iree-preprocessing-pass-pipeline=builtin.module(util.func(iree-global-opt-raise-special-ops, iree-flow-canonicalize), iree-preprocessing-transpose-convolution-pipeline, util.func(iree-preprocessing-pad-to-intrinsics))"
+"--iree-preprocessing-pass-pipeline=builtin.module(util.func(iree-global-opt-raise-special-ops, iree-flow-canonicalize), iree-preprocessing-transpose-convolution-pipeline, util.func(iree-preprocessing-pad-to-intrinsics), util.func(iree-preprocessing-generalize-linalg-matmul-experimental))"
 )
 declare -a FLAGS=("${DEFAULT_FLAGS[*]}")
 
@@ -44,7 +44,6 @@ 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 \