Skip to content

Commit

Permalink
add flag for generalizing batch matmul ops (#52)
Browse files Browse the repository at this point in the history
iree-org/iree#17877

---------

Co-authored-by: Quinn Dawkins <quinn@nod-labs.com>
  • Loading branch information
IanWood1 and qedawkins committed Jul 12, 2024
1 parent 5ed406e commit 7d15e77
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions int8-model/compile-punet-base.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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[*]}")

Expand All @@ -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 \
Expand Down

0 comments on commit 7d15e77

Please sign in to comment.