Skip to content

Commit 8ab1b25

Browse files
ChenhanYuclaude
andcommitted
add: gpu_regression test suite with per-push CI
- Move test_dflash.py to tests/gpu_regression/torch/speculative/ - Add cuda13-gpu-regression tox env - Add gpu-regression matrix entry to gpu_tests.yml (15min timeout) - Runs in parallel with existing gpu/gpu-megatron/gpu-trtllm jobs - Triggers on changes to modelopt/**, tests/gpu_regression/**, examples/speculative_decoding/**, tools/launcher/** Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com> Signed-off-by: Chenhan Yu <chenhany@nvidia.com>
1 parent 2f474aa commit 8ab1b25

File tree

3 files changed

+15
-0
lines changed

3 files changed

+15
-0
lines changed

.github/workflows/gpu_tests.yml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,11 @@ jobs:
4242
.github/workflows/gpu_tests.yml
4343
modelopt/**
4444
tests/gpu/**
45+
tests/gpu_regression/**
46+
examples/speculative_decoding/**
47+
examples/dataset/**
48+
modelopt_recipes/general/speculative_decoding/**
49+
tools/launcher/**
4550
pyproject.toml
4651
tox.ini
4752
fail_on_initial_diff_error: true
@@ -66,6 +71,9 @@ jobs:
6671
timeout: 45
6772
container_image: pytorch:26.01-py3
6873
# tests/gpu/_extensions/test_onnx_extensions.py fails for newer containers until https://github.com/tbenthompson/cppimport/pull/98
74+
- example: gpu-regression
75+
timeout: 15
76+
container_image: pytorch:26.01-py3
6977
- example: gpu-megatron
7078
timeout: 45
7179
container_image: pytorch:26.01-py3
File renamed without changes.

tox.ini

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ envlist=
33
pre-commit-all
44
py312-torch210-tf_latest-unit
55
cuda13-gpu
6+
cuda13-gpu-regression
67
cuda13-gpu-megatron
78
skipsdist = True
89
toxworkdir = /tmp/{env:USER}-modelopt-tox
@@ -68,6 +69,12 @@ commands_pre =
6869
commands =
6970
python -m pytest tests/gpu {env:COV_ARGS:}
7071

72+
[testenv:cuda13-gpu-regression]
73+
commands_pre =
74+
pip install -e .[hf,dev-test]
75+
commands =
76+
python -m pytest tests/gpu_regression {env:COV_ARGS:}
77+
7178
[testenv:cuda13-gpu-megatron]
7279
commands_pre =
7380
# Install deps here so that it gets installed even in --current-env

0 commit comments

Comments
 (0)