From 31635f30f3758a0f31054076e85ad2f40ce9086d Mon Sep 17 00:00:00 2001
From: Mike S Wang <32841762+MikeSWang@users.noreply.github.com>
Date: Wed, 11 Sep 2024 00:46:47 +0100
Subject: [PATCH] BUILD(ci): Modify cross-compilation for CUDA variant

---
 .github/workflows/cd_cuda_xp.yml | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/cd_cuda_xp.yml b/.github/workflows/cd_cuda_xp.yml
index b206e9ae..7468a03f 100644
--- a/.github/workflows/cd_cuda_xp.yml
+++ b/.github/workflows/cd_cuda_xp.yml
@@ -62,6 +62,7 @@ jobs:
             echo "TARGET_ARCH=x86_64" >> "${GITHUB_ENV}"
           elif [[ $(uname -m) == 'x86_64' ]]; then
             echo "TARGET_ARCH=aarch64" >> "${GITHUB_ENV}"
+            echo "TARGET_ARCH_ALT=arm64" >> "${GITHUB_ENV}"
           fi
 
       - name: Checkout (automatic trigger)
@@ -92,7 +93,7 @@ jobs:
         uses: docker/setup-qemu-action@v3
         with:
           # Specify target architecture.
-          platforms: linux/${{ env.TARGET_ARCH }}
+          platforms: linux/${{ env.TARGET_ARCH_ALT }}
 
       - name: Set up Python 3
         uses: actions/setup-python@v5
@@ -114,12 +115,12 @@ jobs:
         env:
           # Specify target architecture, which is actually fixed to aarch64.
           CIBW_ARCHS_LINUX: ${{ env.TARGET_ARCH }}
-          CIBW_MANYLINUX_AARCH64_IMAGE: quay.io/pypa/manylinux_2_24_aarch64
+          CIBW_MANYLINUX_AARCH64_IMAGE: quay.io/pypa/manylinux_2_24_${{ env.TARGET_ARCH }}
           CIBW_BEFORE_ALL_LINUX: >
             apt-get -y install libgsl-devel &&
-            wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/cross-linux-aarch64/cuda-keyring_1.1-1_all.deb &&
+            wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/cross-linux-${{ env.TARGET_ARCH }}/cuda-keyring_1.1-1_all.deb &&
             dpkg -i cuda-keyring_1.1-1_all.deb &&
-            apt-get -y install cuda-cross-aarch64
+            apt-get -y install cuda-cross-${{ env.TARGET_ARCH }}
 
       - name: Verify built distribution
         run: python -m twine check --strict dist/*