cpu: aarch64: support ci with TP

Signed-off-by: Ye Tao <ye.tao@arm.com>

Author: taoye9

.github/automation/build_aarch64.sh

@@ -18,11 +18,48 @@
 # *******************************************************************************
 
 # Build oneDNN for aarch64.
-
 set -o errexit -o pipefail -o noclobber
 
 SCRIPT_DIR="$(dirname "$(readlink -f "$0")")"
 
+
+# Function to clone and install Eigen
+install_eigen() {
+  local eigen_repo="https://gitlab.com/libeigen/eigen.git"
+  local eigen_commit="aa6964bf3a34fd607837dd8123bc42465185c4f8"
+  
+  echo "Cloning Eigen repository..."
+  if ! git clone "$eigen_repo" --depth 1; then
+    echo "Error: Failed to clone Eigen repository."
+    return 1
+  fi
+
+  cd eigen 
+
+  if ! git fetch --depth 1 origin "$eigen_commit" || ! git checkout "$eigen_commit"; then
+    echo "Error: Failed to fetch or checkout commit."
+    return 1
+  fi
+
+  cd ..
+
+  mkdir -p eigen-build && cd eigen-build 
+  
+  if ! cmake -DCMAKE_INSTALL_PREFIX="${EIGEN_INSTALL_PATH}" "$EIGEN_PATH"; then
+    echo "Error: CMake configuration failed."
+    return 1
+  fi
+
+  # Build and install Eigen
+  echo "Building and installing Eigen..."
+  if ! make -s -j install; then
+    echo "Error: Build or installation failed."
+    return 1
+  fi
+
+  echo "Eigen installed successfully!"
+}
+
 # Defines MP, CC, CXX and OS.
 source ${SCRIPT_DIR}/common_aarch64.sh
 
@@ -36,16 +73,32 @@ if [[ "$OS" == "Darwin" ]]; then
     ONEDNN_THREADING=SEQ
 fi
 
-set -x
-cmake \
-    -Bbuild -S. \
+CMAKE_OPTIONS="-Bbuild -S. \
     -DDNNL_AARCH64_USE_ACL=ON \
     -DONEDNN_BUILD_GRAPH=0 \
-    -DDNNL_CPU_RUNTIME=$ONEDNN_THREADING \
     -DONEDNN_WERROR=ON \
     -DDNNL_BUILD_FOR_CI=ON \
-    -DONEDNN_TEST_SET=$ONEDNN_TEST_SET \
-    -DCMAKE_BUILD_TYPE=$CMAKE_BUILD_TYPE
+    -DONEDNN_TEST_SET=${ONEDNN_TEST_SET} \
+    -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE} \
+"
+
+if [[ "$SCHEDULER" == 'TP' ]]; then
+  # install eigen
+  export EIGEN_INSTALL_PATH=${EIGEN_INSTALL_PATH:-"$PWD"}
+  export EIGEN_PATH=${EIGEN_PATH:-"$PWD/eigen"}
+  install_eigen
+
+  CMAKE_OPTIONS="${CMAKE_OPTIONS} -DDNNL_CPU_RUNTIME=THREADPOOL \
+      -D_DNNL_TEST_THREADPOOL_IMPL=EIGEN -DEigen3_DIR=${EIGEN_INSTALL_PATH}/share/eigen3/cmake"
+elif [[ "$SCHEDULER" == "OMP" ] || [ "$SCHEDULER" == "SEQ" ]]; then
+  CMAKE_OPTIONS="${CMAKE_OPTIONS} -DDNNL_CPU_RUNTIME=OMP"
+else
+  echo "Only OMP, TP, SEQ schedulers supported, $SCHEDULER requested"
+  exit 1
+fi
+
+set -x
+cmake ${CMAKE_OPTIONS}
 
 cmake --build build $MP
 set +x

.github/automation/build_acl.sh

@@ -36,7 +36,7 @@ if [[ "$OS" == "Linux" ]]; then
     ACL_MULTI_ISA_SUPPORT=1
     if [[ "$ACL_THREADING" == "OMP" ]]; then
         ACL_OPENMP=1
-    elif [[ "$ACL_THREADING" == "SEQ" ]]; then
+    elif [[ "$ACL_THREADING" == "SEQ" || "$ACL_THREADING" == "TP" ]]; then
         ACL_OPENMP=0
     fi
     ACL_OS="linux"

.github/workflows/ci-aarch64.yml

@@ -70,27 +70,31 @@ jobs:
           CMAKE_BUILD_TYPE: ${{ matrix.config }}
           GCC_VERSION: 14
 
-      - if: matrix.toolset == 'clang'
-        name: Run oneDNN smoke tests
+      - name: Run oneDNN smoke tests
         run: ${{ github.workspace }}/oneDNN/.github/automation/test_aarch64.sh
         working-directory: ${{ github.workspace }}/oneDNN/build
         env:
           CMAKE_BUILD_TYPE: ${{ matrix.config }}
           DYLD_LIBRARY_PATH: ${{ github.workspace }}/ComputeLibrary/build
-
-  # We only run the linux aarch64 runners if macos smoke tests pass.
+    
+  # We only run the linux aarch64 runners if macOS smoke tests pass.
   linux:
     needs: macos
     strategy:
       matrix:
-        threading: [OMP]
         toolset: [clang, gcc]
         config: [Debug, Release]
-        host: [
-          { name: c6g, label: ah-ubuntu_22_04-c6g_2x-50 }, 
-          { name: c7g, label: ah-ubuntu_22_04-c7g_2x-50 }
-        ]
-
+        threading: [ OMP ]
+        host: [ { name: c6g, label: ah-ubuntu_22_04-c6g_2x-50 }, { name: c7g, label: ah-ubuntu_22_04-c7g_2x-50 }]
+        include:
+          - toolset: clang
+            config: Release
+            threading: TP
+            host: { name: c6g, label: ah-ubuntu_22_04-c6g_2x-50 }
+          - toolset: clang
+            config: Release
+            threading: TP
+            host: { name: c7g, label: ah-ubuntu_22_04-c7g_2x-50 }
     name: ${{ matrix.host.name }}, ${{ matrix.toolset }}, ${{ matrix.threading }}, ${{ matrix.config }}
     runs-on: ${{ matrix.host.label }}
     steps:
@@ -140,6 +144,8 @@ jobs:
           CMAKE_BUILD_TYPE: ${{ matrix.config }}
           GCC_VERSION: 13
           ONEDNN_THREADING: ${{ matrix.threading }}
+          EIGEN_INSTALL_PATH: ${{ github.workspace }}
+          SCHEDULER: ${{ matrix.threading }}
 
       - name: Run oneDNN tests
         run: ${{ github.workspace }}/oneDNN/.github/automation/test_aarch64.sh
@@ -148,4 +154,4 @@ jobs:
           BUILD_TOOLSET: ${{ matrix.toolset }}
           CMAKE_BUILD_TYPE: ${{ matrix.config }}
           DYLD_LIBRARY_PATH: ${{ github.workspace }}/ComputeLibrary/build
-          ONEDNN_THREADING: ${{ matrix.threading }}     
+          ONEDNN_THREADING: ${{ matrix.threading }}