From 5ed406e01d4d8d9410f5ea929e1538cff76740a6 Mon Sep 17 00:00:00 2001 From: Lei Zhang Date: Fri, 12 Jul 2024 15:16:49 +0000 Subject: [PATCH] [int8] Add option to generalize matmul ops --- int8-model/compile-punet-base.sh | 1 + 1 file changed, 1 insertion(+) 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 \