From 29ce7cda6f7b9e31802d38f2f908019b0fa26787 Mon Sep 17 00:00:00 2001
From: fanqiNO1 <1848839264@qq.com>
Date: Mon, 20 Nov 2023 17:17:00 +0800
Subject: [PATCH] [Fix] Add cu118 test

---
 .github/workflows/merge_stage_test.yml | 2 +-
 .github/workflows/pr_stage_test.yml    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/merge_stage_test.yml b/.github/workflows/merge_stage_test.yml
index c5456e28ac..23878a3c5c 100644
--- 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
index 7127c4644c..b2576cfd57 100644
--- 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' }}