setup aarch64 g++ compiler

bitsandbytes-foundation · Feb 6, 2024 · b121116 · b121116
1 parent b397cbb
commit b121116
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -49,7 +49,7 @@ jobs:
       if: ${{ matrix.os == 'ubuntu-latest' && matrix.arch == 'aarch64' }}
       run: |
         # Allow cross-compile on aarch64
-        sudo apt-get install -y gcc-aarch64-linux-gnu binutils-aarch64-linux-gnu
+        sudo apt-get install -y g++-aarch64-linux-gnu binutils-aarch64-linux-gnu
 
     - name: Setup cmake
       uses: jwlawson/[email protected]
@@ -79,6 +79,8 @@ jobs:
         python3 -m pip install cmake==3.27.9 ninja
         if [ "${{ matrix.os }}" = "windows-latest" ]; then
           echo CXX_COMPILER=cl >> "$GITHUB_ENV"
+        elif [ "${{ matrix.os }}" = "ubuntu-latest" -a "${{ matrix.arch }}" == "aarch64" ]; then
+          echo CXX_COMPILER=aarch64-linux-gnu-g++ >> "$GITHUB_ENV"
         else
           echo CXX_COMPILER=g++ >> "$GITHUB_ENV"
         fi
@@ -176,7 +178,7 @@ jobs:
       if: ${{ matrix.os == 'ubuntu-latest' && matrix.arch == 'aarch64' }}
       run: |
         # Allow cross-compile on aarch64
-        sudo apt-get install -y gcc-aarch64-linux-gnu binutils-aarch64-linux-gnu
+        sudo apt-get install -y g++-aarch64-linux-gnu binutils-aarch64-linux-gnu
 
     - name: CUDA Toolkit
       shell: bash -el {0}
@@ -205,6 +207,8 @@ jobs:
             echo CXX_COMPILER=cl >> "$GITHUB_ENV"
             # without -DCMAKE_CUDA_COMPILER=nvcc, cmake config always fail for cuda-11.8
             echo DCMAKE_CUDA_COMPILER=-DCMAKE_CUDA_COMPILER=nvcc >> "$GITHUB_ENV"
+        elif [ "${{ matrix.os }}" = "ubuntu-latest" -a "${{ matrix.arch }}" == "aarch64" ]; then
+          echo CXX_COMPILER=aarch64-linux-gnu-g++ >> "$GITHUB_ENV"
         else
             echo CXX_COMPILER=g++ >> "$GITHUB_ENV"
         fi
@@ -309,7 +313,7 @@ jobs:
       if: ${{ matrix.os == 'ubuntu-latest' && matrix.arch == 'aarch64' }}
       run: |
         # Allow cross-compile on aarch64
-        sudo apt-get install -y gcc-aarch64-linux-gnu binutils-aarch64-linux-gnu
+        sudo apt-get install -y g++-aarch64-linux-gnu binutils-aarch64-linux-gnu
 
     - name: CUDA Toolkit
       shell: bash -el {0}
@@ -321,7 +325,11 @@ jobs:
         CUDA_HOME="${{ env.CONDA }}/envs/bnb-env"
         echo CUDA_HOME=$CUDA_HOME >> "$GITHUB_ENV"
         echo CUDA_PATH=$CUDA_HOME >> "$GITHUB_ENV"
-        echo CXX_COMPILER=g++ >> "$GITHUB_ENV"
+        if [ "${{ matrix.os }}" = "ubuntu-latest" -a "${{ matrix.arch }}" == "aarch64" ]; then
+          echo CXX_COMPILER=aarch64-linux-gnu-g++ >> "$GITHUB_ENV"
+        else
+          echo CXX_COMPILER=g++ >> "$GITHUB_ENV"
+        fi
 
         nvcc --version