From 4f097713bbaa2efaf30cd2d5864dd60762acd924 Mon Sep 17 00:00:00 2001
From: CSY-ModelCloud <csy@modelcloud.ai>
Date: Thu, 16 Apr 2026 20:56:52 +0800
Subject: [PATCH 1/2] Update unit_tests.yml

---
 .github/workflows/unit_tests.yml | 32 ++++++++++++++++++++++++++++----
 1 file changed, 28 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index 398d2463b..167569ad6 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -183,8 +183,8 @@ jobs:
         - /monster/ci/models:/monster/data/model
         - /monster/ci/dataset:/monster/data/model/dataset
         - /monster/ci/huggingface:/github/home/.cache/huggingface
-        # - /monster/ci/uv:/opt/uv
         - /github/workspace/uv:/opt/uv
+        - /github/workspace/tmp:/opt/uv/tmp
         - /monster/ci/uv/python:/opt/uv/python
         - /monster/ci/uv/cache/python:/opt/uv/cache/python
         - /monster/ci/uv/setup_uv_venv.sh:/opt/uv/setup_uv_venv.sh
@@ -279,7 +279,10 @@ jobs:
           echo "-- loaded --"
 
           echo "-- setting up env --"
-          env_name="gptqmodel_test_cu${{ needs.check-vm.outputs.cuda_version }}_torch${{ env.TORCH_VERSION }}_py${py}_${{ matrix.test_script }}"
+          safe_ref=$(printf '%s' "${{ env.ref }}" | sed -e 's/[\\/]/_/g' -e 's/^refs_heads_//')
+          env_name="gptqmodel_$safe_ref_test_${{ matrix.test_script }}"
+          # will clean later
+          mv /opt/uv/venvs/$env_name "/opt/uv/tmp/${env_name}_$(date +%s)" || true
           /opt/uv/setup_uv_venv.sh $env_name
           echo "-- set --"
 
@@ -467,8 +470,8 @@ jobs:
         - /monster/ci/models:/monster/data/model
         - /monster/ci/dataset:/monster/data/model/dataset
         - /monster/ci/huggingface:/github/home/.cache/huggingface
-        # - /monster/ci/uv:/opt/uv
         - /github/workspace/uv:/opt/uv
+        - /github/workspace/tmp:/opt/uv/tmp
         - /monster/ci/uv/python:/opt/uv/python
         - /monster/ci/uv/cache/python:/opt/uv/cache/python
         - /monster/ci/uv/setup_uv_venv.sh:/opt/uv/setup_uv_venv.sh
@@ -546,8 +549,29 @@ jobs:
 
       - name: Activate uv env
         run: |
-          env_name="gptqmodel_test_cu${{ needs.check-vm.outputs.cuda_version }}_torch${{ env.TORCH_VERSION }}_py${{ env.PYTHON_VERSION }}_${{ matrix.test_script }}"
+          echo "-- loading unit test's config --"
+          source /opt/uv/setup_uv_venv.sh unit_test_env
+
+          config_json="$(python3 .github/scripts/parse_test_config.py \
+            --group tests \
+            --test-name "${{ matrix.test_script }}")"
+
+          py="$(printf '%s' "$config_json" | python3 -c 'import json, sys; print(json.load(sys.stdin)["py"])')"
+          gpu="$(printf '%s' "$config_json" | python3 -c 'import json, sys; print(json.load(sys.stdin)["gpu"])')"
+
+          echo "PYTHON_VERSION=$py" >> "$GITHUB_ENV"
+          echo "GPU_COUNT=$gpu" >> "$GITHUB_ENV"
+          
+          echo "using py=$py gpu=$gpu for test ${{ matrix.test_script }}"
+          echo "-- loaded --"
+
+          echo "-- setting up env --"
+          safe_ref=$(printf '%s' "${{ env.ref }}" | sed -e 's/[\\/]/_/g' -e 's/^refs_heads_//')
+          env_name="gptqmodel_$safe_ref_test_${{ matrix.test_script }}"
+          # will clean later
+          mv /opt/uv/venvs/$env_name "/opt/uv/tmp/${env_name}_$(date +%s)" || true
           /opt/uv/setup_uv_venv.sh $env_name
+          echo "-- set --"
 
       - name: Setup uv env
         run: |

From d39194c11ee8fa1355450df049cb5370d0d28833 Mon Sep 17 00:00:00 2001
From: CSY-ModelCloud <csy@modelcloud.ai>
Date: Thu, 16 Apr 2026 21:01:25 +0800
Subject: [PATCH 2/2] print env name

---
 .github/workflows/unit_tests.yml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index 167569ad6..d6fb11b15 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -280,7 +280,8 @@ jobs:
 
           echo "-- setting up env --"
           safe_ref=$(printf '%s' "${{ env.ref }}" | sed -e 's/[\\/]/_/g' -e 's/^refs_heads_//')
-          env_name="gptqmodel_$safe_ref_test_${{ matrix.test_script }}"
+          env_name="gptqmodel_${safe_ref}_test_${{ matrix.test_script }}"
+          echo "env_name: $env_name"
           # will clean later
           mv /opt/uv/venvs/$env_name "/opt/uv/tmp/${env_name}_$(date +%s)" || true
           /opt/uv/setup_uv_venv.sh $env_name
@@ -567,7 +568,8 @@ jobs:
 
           echo "-- setting up env --"
           safe_ref=$(printf '%s' "${{ env.ref }}" | sed -e 's/[\\/]/_/g' -e 's/^refs_heads_//')
-          env_name="gptqmodel_$safe_ref_test_${{ matrix.test_script }}"
+          env_name="gptqmodel_${safe_ref}_test_${{ matrix.test_script }}"
+          echo "env_name: $env_name"
           # will clean later
           mv /opt/uv/venvs/$env_name "/opt/uv/tmp/${env_name}_$(date +%s)" || true
           /opt/uv/setup_uv_venv.sh $env_name