diff --git a/.github/workflows/macos-cpu-wheels.yml b/.github/workflows/macos-cpu-wheels.yml index e08ef24..bb7ca38 100644 --- a/.github/workflows/macos-cpu-wheels.yml +++ b/.github/workflows/macos-cpu-wheels.yml @@ -26,14 +26,17 @@ jobs: id: set-matrix run: | # outputting for debugging purposes - python ./scripts/github_actions/generate_build_matrix.py --for-macos - MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-macos) + # python ./scripts/github_actions/generate_build_matrix.py --for-macos + # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-macos) + # + python ./scripts/github_actions/generate_build_matrix.py --for-macos --test-only-latest-torch + MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-macos --test-only-latest-torch) echo "::set-output name=matrix::${MATRIX}" build_wheels_macos_cpu: needs: generate_build_matrix name: ${{ matrix.torch }} ${{ matrix.python-version }} - runs-on: macos-11 + runs-on: macos-latest strategy: fail-fast: false matrix: @@ -47,7 +50,7 @@ jobs: # see https://cibuildwheel.readthedocs.io/en/stable/changelog/ # for a list of versions - name: Build wheels - uses: pypa/cibuildwheel@v2.11.4 + uses: pypa/cibuildwheel@v2.16.5 env: CIBW_BEFORE_BUILD: pip install torch==${{ matrix.torch}} cmake numpy CIBW_BUILD: ${{ matrix.python-version }}-* diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml index 44acd79..66c9965 100644 --- a/.github/workflows/ubuntu-cuda-wheels.yml +++ b/.github/workflows/ubuntu-cuda-wheels.yml @@ -26,8 +26,11 @@ jobs: id: set-matrix run: | # outputting for debugging purposes - python ./scripts/github_actions/generate_build_matrix.py --enable-cuda - MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda) + # python ./scripts/github_actions/generate_build_matrix.py --enable-cuda + # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda) + + python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch + MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch) echo "::set-output name=matrix::${MATRIX}" build-manylinux-wheels: diff --git a/scripts/github_actions/build-ubuntu-cuda.sh b/scripts/github_actions/build-ubuntu-cuda.sh index 56347f5..526f915 100755 --- a/scripts/github_actions/build-ubuntu-cuda.sh +++ b/scripts/github_actions/build-ubuntu-cuda.sh @@ -22,6 +22,10 @@ if [ -z $CUDA_VERSION ]; then exit 1 fi +if [[ $TORCH_VERSION =~ 2.2.* && $CUDA_VERSION =~ 12.* ]]; then + export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0" +fi + yum -y install openssl-devel bzip2-devel libffi-devel xz-devel wget redhat-lsb-core