Build macos wheel for torch2.0.0 (#3008)

pull/3024/head
Zaida Zhou 2023-12-27 10:43:08 +08:00 committed by GitHub
parent 1bb3dea9ad
commit 5234915c5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 5 deletions

View File

@ -11,12 +11,12 @@ env:
jobs:
build_macos10_wheel:
runs-on: macos-10.15
runs-on: macos-latest
if: contains(github.event.head_commit.message, 'Bump version to')
strategy:
matrix:
torch: [1.5.1, 1.6.0, 1.7.0, 1.8.0, 1.9.0, 1.10.0, 1.11.0, 1.12.0, 1.13.0]
python-version: [3.6, 3.7, 3.8, 3.9, '3.10']
torch: [1.5.1, 1.6.0, 1.7.0, 1.8.0, 1.9.0, 1.10.0, 1.11.0, 1.12.0, 1.13.0, 2.0.0]
python-version: [3.6, 3.7, 3.8, 3.9, '3.10', '3.11']
include:
- torch: 1.5.1
torchvision: 0.6.1
@ -36,31 +36,55 @@ jobs:
torchvision: 0.13.0
- torch: 1.13.0
torchvision: 0.14.0
- torch: 2.0.0
torchvision: 0.15.1
exclude:
- torch: 1.5.1
python-version: 3.9
- torch: 1.5.1
python-version: '3.10'
- torch: 1.5.1
python-version: '3.11'
- torch: 1.6.0
python-version: 3.9
- torch: 1.6.0
python-version: '3.10'
- torch: 1.6.0
python-version: '3.11'
- torch: 1.7.0
python-version: 3.9
- torch: 1.7.0
python-version: '3.10'
- torch: 1.7.0
python-version: '3.11'
- torch: 1.8.0
python-version: '3.10'
- torch: 1.8.0
python-version: '3.11'
- torch: 1.9.0
python-version: '3.10'
- torch: 1.9.0
python-version: '3.11'
- torch: 1.10.0
python-version: '3.10'
- torch: 1.10.0
python-version: '3.11'
- torch: 1.11.0
python-version: 3.6
- torch: 1.11.0
python-version: '3.11'
- torch: 1.12.0
python-version: 3.6
- torch: 1.12.0
python-version: '3.11'
- torch: 1.13.0
python-version: 3.6
- torch: 1.13.0
python-version: '3.11'
- torch: 2.0.0
python-version: 3.6
- torch: 2.0.0
python-version: 3.7
steps:
- uses: actions/checkout@v2
- name: Set up Python

View File

@ -2,10 +2,9 @@
import numpy as np
import pytest
import torch
from mmengine.utils import digit_version
from mmcv.utils import (IS_CUDA_AVAILABLE, IS_MLU_AVAILABLE, IS_MPS_AVAILABLE,
IS_NPU_AVAILABLE)
IS_NPU_AVAILABLE, digit_version)
class TestBBox: