From 5234915c5efda6d080223be6ff6606874876e901 Mon Sep 17 00:00:00 2001
From: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Date: Wed, 27 Dec 2023 10:43:08 +0800
Subject: [PATCH] Build macos wheel for torch2.0.0 (#3008)

---
 .github/workflows/build_macos_wheel.yml | 30 ++++++++++++++++++++++---
 tests/test_ops/test_bbox.py             |  3 +--
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/build_macos_wheel.yml b/.github/workflows/build_macos_wheel.yml
index b1d09628f..ce468529f 100644
--- a/.github/workflows/build_macos_wheel.yml
+++ b/.github/workflows/build_macos_wheel.yml
@@ -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
diff --git a/tests/test_ops/test_bbox.py b/tests/test_ops/test_bbox.py
index 752d87766..c888184e7 100644
--- a/tests/test_ops/test_bbox.py
+++ b/tests/test_ops/test_bbox.py
@@ -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: