[ZENDNN FWK] Updated Env Variable for MATMUL

1. SGEMM matmul along with env variable ZENDNN_GEMM_ALGO is removed 2. New env variable for setting MatMul algo which uses BRGEMM export ZENDNN_MATMUL_ALGO=FP32:4 export ZENDNN_MATMUL_ALGO=BF16:3 Signed-off-by: Devasena Kirthika S <[email protected]> Change-Id: Ied8c908a206c8ec3e387e06275a65cb3acacc10f Reviewed-on: https://gerrit-git.amd.com/c/amd/ec/ZenDNN_ONNXRuntime/+/1033972 Tested-by: Llvm Cpu <[email protected]> Reviewed-by: Chandra Kumar Ramasamy <[email protected]> (cherry picked from commit 7ed80c7) Reviewed-on: https://gerrit-git.amd.com/c/amd/ec/ZenDNN_ONNXRuntime/+/1034266 (cherry picked from commit 64511fb730b682c4d8f3753b29b3d55940d59499) Reviewed-on: https://gerrit-git.amd.com/c/amd/ec/ZenDNN_ONNXRuntime/+/1036384 Reviewed-by: Ajeet Singh <[email protected]>
amd · Apr 12, 2024 · 964219a · 964219a
1 parent d36916f
commit 964219a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/scripts/zendnn_ONNXRT_env_setup.sh b/scripts/zendnn_ONNXRT_env_setup.sh
@@ -188,9 +188,9 @@ echo "ZENDNN_INT8_SUPPORT=$ZENDNN_INT8_SUPPORT"
 export ZENDNN_RELU_UPPERBOUND=0
 echo "ZENDNN_RELU_UPPERBOUND=$ZENDNN_RELU_UPPERBOUND"
 
-# ZENDNN_GEMM_ALGO is set to 3 by default
-export ZENDNN_GEMM_ALGO=3
-echo "ZENDNN_GEMM_ALGO=$ZENDNN_GEMM_ALGO"
+# ZENDNN_MATMUL_ALGO is set to FP32:3 and BF16:3 by default
+export ZENDNN_MATMUL_ALGO=FP32:3,BF16:3
+echo "ZENDNN_MATMUL_ALGO=$ZENDNN_MATMUL_ALGO"
 
 #Use local copy of ZenDNN library source code when building ONNXRT
 #Default is build from local source for development and verification.

diff --git a/scripts/zendnn_ONNXRT_env_setup_win.bat b/scripts/zendnn_ONNXRT_env_setup_win.bat
@@ -67,9 +67,9 @@ echo ZENDNN_INT8_SUPPORT=%ZENDNN_INT8_SUPPORT%
 set ZENDNN_RELU_UPPERBOUND=0
 echo ZENDNN_RELU_UPPERBOUND=%ZENDNN_RELU_UPPERBOUND%
 
-::ZENDNN_GEMM_ALGO is set to 3 by default
-set ZENDNN_GEMM_ALGO=4
-echo ZENDNN_GEMM_ALGO=%ZENDNN_GEMM_ALGO%
+::ZENDNN_MATMUL_ALGO is set to FP32:3 and BF16:3 by default
+set ZENDNN_MATMUL_ALGO=FP32:3,BF16:3
+echo ZENDNN_MATMUL_ALGO=%ZENDNN_MATMUL_ALGO%
 :: Use local copy of ZenDNN library source code when building ONNXRT
 :: Default is build from local source for development and verification.
 :: For release, set ZENDNN_ONNXRT_USE_LOCAL_ZENDNN=0