Skip to content

Commit

Permalink
BUILD(ci): Modify cross-compilation for CUDA variant
Browse files Browse the repository at this point in the history
  • Loading branch information
MikeSWang committed Sep 10, 2024
1 parent 633c451 commit 31635f3
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions .github/workflows/cd_cuda_xp.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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
Expand All @@ -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/*
Expand Down

0 comments on commit 31635f3

Please sign in to comment.