f

Change-Id: I9ed6fd6eaa27050abf196068e9e49e0912227927

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
 
@@ -46,7 +83,11 @@ CMAKE_OPTIONS="-Bbuild -S. \
 "
 
 if [[ "$SCHEDULER" == 'TP' ]]; then
-  EIGEN_INSTALL_PATH=${EIGEN_INSTALL_PATH:-"$PWD"}
+  # 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

.github/workflows/ci-aarch64.yml

@@ -122,19 +122,6 @@ jobs:
         with:
           version: "17"
 
-      - if: matrix.threading == 'TP'
-        name: Install eigen3
-        working-directory: ${{ github.workspace }}
-        env:
-          COMPILER: ${{ matrix.toolset }} 
-        run: |
-          git clone https://gitlab.com/libeigen/eigen.git --depth 1
-          cd eigen
-          git fetch --depth 1 origin aa6964bf3a34fd607837dd8123bc42465185c4f8
-          git checkout aa6964bf3a34fd607837dd8123bc42465185c4f8
-          cd ..
-          ${{ github.workspace }}/oneDNN/.github/automation/build-eigen-headers.sh
-
       - name: Build ACL
         run: ${{ github.workspace }}/oneDNN/.github/automation/build_acl.sh
         env: