diff --git a/.github/workflows/macos-cpu-wheels.yml b/.github/workflows/macos-cpu-wheels.yml index e08ef24..62ae611 100644 --- a/.github/workflows/macos-cpu-wheels.yml +++ b/.github/workflows/macos-cpu-wheels.yml @@ -28,12 +28,15 @@ jobs: # 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 --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: @@ -44,15 +47,22 @@ jobs: with: fetch-depth: 0 - # see https://cibuildwheel.readthedocs.io/en/stable/changelog/ - # for a list of versions - - name: Build wheels - uses: pypa/cibuildwheel@v2.11.4 - env: - CIBW_BEFORE_BUILD: pip install torch==${{ matrix.torch}} cmake numpy - CIBW_BUILD: ${{ matrix.python-version }}-* - CIBW_REPAIR_WHEEL_COMMAND_MACOS: "" - CIBW_BUILD_VERBOSITY: 3 + - name: Setup Python + uses: actions/setup-python@v2 + with: + python-version: ${{ matrix.python-version }} + + - name: Install dependencies + shell: bash + run: | + pip install -q torch==${{ matrix.torch}} cmake numpy wheel twine setuptools + + - name: Build wheel + shell: bash + run: | + python3 setup.py bdist_wheel + mkdir wheelhouse + cp -v dist/* wheelhouse - name: Display wheels shell: bash diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml index 44acd79..82b5721 100644 --- a/.github/workflows/ubuntu-cuda-wheels.yml +++ b/.github/workflows/ubuntu-cuda-wheels.yml @@ -28,6 +28,9 @@ jobs: # 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 --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..c3d1423 100755 --- a/scripts/github_actions/build-ubuntu-cuda.sh +++ b/scripts/github_actions/build-ubuntu-cuda.sh @@ -22,6 +22,11 @@ if [ -z $CUDA_VERSION ]; then exit 1 fi +if [[ $TORCH_VERSION =~ 2.2.* && $CUDA_VERSION =~ 12.* ]]; then + # see https://github.com/pytorch/pytorch/issues/113948 + 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 diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py index 85f9541..d747afd 100755 --- a/scripts/github_actions/generate_build_matrix.py +++ b/scripts/github_actions/generate_build_matrix.py @@ -211,9 +211,11 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_ if p in excluded_python_versions: continue - if for_windows or for_macos: + if for_windows: p = "cp" + "".join(p.split(".")) ans.append({"torch": torch, "python-version": p}) + elif for_macos: + ans.append({"torch": torch, "python-version": p}) else: ans.append( {