Skip to content

Commit

Permalink
BUG: Use newer CI packaging to exclude libraries
Browse files Browse the repository at this point in the history
  • Loading branch information
Simon Rit committed Feb 14, 2024
1 parent 6ba7b46 commit bdc9b37
Showing 1 changed file with 9 additions and 5 deletions.
14 changes: 9 additions & 5 deletions .github/workflows/build-test-package-python-cuda.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@ on: [push,pull_request]
env:
cmake-options: '-DCUDACOMMON_CUDA_VERSION=11.6'
itk-wheel-tag: 'v5.4rc02'
itk-python-package-tag: '5ad02309321621cdc7269b9b68a35013c912271c'
itk-python-package-org: 'InsightSoftwareConsortium'
itk-python-package-tag: '61c2757664821edf0ebcf8cabd0af66de4f87961'
itk-python-package-org: 'SimonRit'

jobs:
build-linux-cuda-python-packages:
Expand All @@ -31,13 +31,17 @@ jobs:
shell: bash
run: |
rm -rf dist
export ITK_PACKAGE_VERSION=${{ env.itk-wheel-tag }}
export ITKPYTHONPACKAGE_TAG=${{ env.itk-python-package-tag }}
export ITKPYTHONPACKAGE_ORG=${{ env.itk-python-package-org }}
export ITK_MODULE_PREQ=${{ env.itk-module-deps }}
CMAKE_OPTIONS=(--cmake_options "${{ env.cmake-options }} -DCUDAToolkit_ROOT=/usr/lib64/cuda116 -DCMAKE_CUDA_COMPILER=/usr/lib64/cuda116/bin/nvcc")
export LD_LIBRARY_PATH="/home/srit/Downloads/cuda116:/home/srit/Downloads/cuda116/targets/x86_64-linux/lib:/home/srit/Downloads/cuda116/lib64/stubs"
# The first two are not library paths but are included to be mounted in the
# docker by dockcross-manylinux-build-module-wheels.sh
export LD_LIBRARY_PATH="/home/srit/Downloads/cuda116"
export LD_LIBRARY_PATH="/usr/lib64/libcuda.so.1:$LD_LIBRARY_PATH"
export LD_LIBRARY_PATH="/home/srit/Downloads/cuda116/targets/x86_64-linux/lib::$LD_LIBRARY_PATH"
if test -e ../../ITKPythonBuilds-linux-manylinux2014.tar.zst ; then
mv ../../*zst .
fi
Expand All @@ -53,7 +57,7 @@ jobs:
export ITK_MODULE_NO_CLEANUP=TRUE
echo "Building for manylinux specialization ${MANYLINUX_VERSION} and target architecture ${TARGET_ARCH}"
./dockcross-manylinux-download-cache-and-build-module-wheels.sh "${CMAKE_OPTIONS[@]}" -x "libcuda.so;libcuda.so.1;libcudart.so;libcudart.so.11.0;libcublas.so;libcublas.so.11;libcublasLt.so;libcublasLt.so.11;libcufft.so;libcufft.so.10" cp3${{ matrix.python3-minor-version }}
./dockcross-manylinux-download-cache-and-build-module-wheels.sh "${CMAKE_OPTIONS[@]}" -x "libcuda.so;libcuda.so.1;libcuda.so.510.39.01;libcudart.so;libcudart.so.11.0;.so.11.6.55" cp3${{ matrix.python3-minor-version }}
mv *zst ../..
- name: Validate build output
Expand Down

0 comments on commit bdc9b37

Please sign in to comment.