change scheduler to ONEDNN_THREADING

Change-Id: I8789154698a44c123ff186624240570185187978
taoye9 · Oct 16, 2024 · 5ddede6 · 5ddede6
1 parent d5ead02
commit 5ddede6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/automation/build_aarch64.sh b/.github/automation/build_aarch64.sh
@@ -82,7 +82,7 @@ CMAKE_OPTIONS="-Bbuild -S. \
     -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE} \
 "
 
-if [[ "$SCHEDULER" == 'TP' ]]; then
+if [[ "$ONEDNN_THREADING" == 'TP' ]]; then
   # install eigen
   export EIGEN_INSTALL_PATH=${EIGEN_INSTALL_PATH:-"$PWD"}
   export EIGEN_PATH=${EIGEN_PATH:-"$PWD/eigen"}
@@ -93,7 +93,7 @@ if [[ "$SCHEDULER" == 'TP' ]]; then
 elif [[ "$SCHEDULER" == "OMP" ] || [ "$SCHEDULER" == "SEQ" ]]; then
   CMAKE_OPTIONS="${CMAKE_OPTIONS} -DDNNL_CPU_RUNTIME=OMP"
 else
-  echo "Only OMP, TP, SEQ schedulers supported, $SCHEDULER requested"
+  echo "Only OMP, TP, SEQ schedulers supported, $ONEDNN_THREADING requested"
   exit 1
 fi