Skip to content

Commit cf8eda2

Browse files
[CI] use torch 2.12.0 & python 3.14t as default on CI (#2894)
* fix command got wrong args * [CI] use torch 2.12.0 & py 3.14t as default * [CI] install torchao * [CI] fix path * [CI] detail log
1 parent ddb23e7 commit cf8eda2

3 files changed

Lines changed: 16 additions & 14 deletions

File tree

.github/scripts/test.yaml

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,15 +7,11 @@ common:
77
tests:
88
test_multi_gpu_inference: # per-test config
99
gpu: 5
10-
py: 3.14t
1110

1211
test_calibration_data_device:
1312
gpu: 2
14-
py: 3.14t
1513

1614
tests/models:
17-
py: 3.13
18-
1915
test_llama3_2_fp8:
2016
sm: '12.0'
2117

.github/workflows/unit_tests.yml

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -67,9 +67,8 @@ env:
6767
CUDA_VERSION: 130
6868
UV_TORCH_BACKEND: cu130
6969
TORCH_VERSION: 2.11.0
70-
# vllm doesn't support 3.14
71-
PYTHON_VERSION: 3.13
72-
UV_PYTHON: 3.13
70+
PYTHON_VERSION: 3.14t
71+
UV_PYTHON: 3.14t
7372
# PYTHON_GIL: 0 // test libs don't support yet
7473
IGNORED_TEST_FILES: "test_tgi.py,test_gptneox.py,models/test_mixtral.py,models/test_phi_3_moe.py,test_bits_new.py,models/test_internlm.py,models/test_internlm2_5.py,models/test_xverse.py"
7574
MODEL_TEST_MODE: ${{ github.event.inputs['model-test-mode'] || 'fast' }}
@@ -247,19 +246,19 @@ jobs:
247246
- name: Test pypi pip
248247
env_name: pypi_pip_env_${{ github.run_id }}_${{ github.run_attempt }}
249248
use_pip: 'true'
250-
install_cmd: pip install gptqmodel torch -U
249+
install_cmd: pip install gptqmodel torch -U -vvv
251250
- name: Test pypi uv
252251
env_name: pypi_uv_env_${{ github.run_id }}_${{ github.run_attempt }}
253252
use_pip: 'false'
254-
install_cmd: uv pip install gptqmodel torch -U
253+
install_cmd: uv pip install gptqmodel torch -U -vvv
255254
- name: Test local pip
256255
env_name: local_pip_env_${{ github.run_id }}_${{ github.run_attempt }}
257256
use_pip: 'true'
258-
install_cmd: pip install . torch -U
257+
install_cmd: pip install . torch -U -vvv
259258
- name: Test local uv
260259
env_name: local_uv_env_${{ github.run_id }}_${{ github.run_attempt }}
261260
use_pip: 'false'
262-
install_cmd: uv pip install . torch -U
261+
install_cmd: uv pip install . torch -U -vvv
263262
steps:
264263
- name: Checkout Codes
265264
uses: actions/checkout@v6
@@ -278,6 +277,13 @@ jobs:
278277
if [[ "${{ matrix.use_pip }}" == "true" ]]; then
279278
uv pip install pip -U
280279
fi
280+
if python -VV 2>&1 | grep -q '3\.14.*free-threading'; then
281+
echo "[INFO] Python 3.14t detected, installing torchao..."
282+
uv pip install http://10.0.13.31/files/torchao-0.18.0+git13cd013d6-cp314-cp314t-linux_x86_64.whl
283+
else
284+
echo "[INFO] Current Python is not 3.14t, skip torchao."
285+
fi
286+
281287
${{ matrix.install_cmd }}
282288
283289

.github/workflows/unit_tests_reusable.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,9 @@ env:
5858
GPU_ALLOCATOR_URL: http://10.0.13.31/gpu
5959
LOGBAR_ANIMATION: '0'
6060
UV_TORCH_BACKEND: cu130
61-
TORCH_VERSION: 2.11.0
62-
PYTHON_VERSION: 3.13
63-
UV_PYTHON: 3.13
61+
TORCH_VERSION: 2.12.0
62+
PYTHON_VERSION: 3.14t
63+
UV_PYTHON: 3.14t
6464
MODEL_TEST_MODE: ${{ inputs.model_test_mode || 'fast' }}
6565
MODEL_TEST_GPU_COUNT: '1'
6666
DEFUSER_GIT_URL: git+https://github.com/modelcloud/Defuser.git

0 commit comments

Comments
 (0)