[Fix] Add cu118 test

open-mmlab · Nov 20, 2023 · 29ce7cd · 29ce7cd
1 parent ae09670
commit 29ce7cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/merge_stage_test.yml b/.github/workflows/merge_stage_test.yml
@@ -312,4 +312,4 @@ jobs:
       - name: Run GPU unittests
         # Skip testing distributed related unit tests since the memory of windows CI is limited
         run: pytest tests/ --ignore tests/test_dist --ignore tests/test_optim/test_optimizer/test_optimizer_wrapper.py --ignore tests/test_model/test_wrappers/test_model_wrapper.py
-        if: ${{ matrix.platform == 'cu111' }}
+        if: ${{ matrix.platform == 'cu111' }} || ${{ matrix.platform == 'cu118' }}
diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml
@@ -169,4 +169,4 @@ jobs:
       - name: Run GPU unittests
         # Skip testing distributed related unit tests since the memory of windows CI is limited
         run: pytest tests/ --ignore tests/test_dist --ignore tests/test_optim/test_optimizer/test_optimizer_wrapper.py --ignore tests/test_model/test_wrappers/test_model_wrapper.py --ignore tests/test_hooks/test_sync_buffers_hook.py
-        if: ${{ matrix.platform == 'cu111' }}
+        if: ${{ matrix.platform == 'cu111' }} || ${{ matrix.platform == 'cu118' }}