Skip to content

Commit 06bb7ff

Browse files
sayak@huggingface.coclaude
andcommitted
[ci] select nightly quantization tests by marker across test tiers
Each backend job now runs `pytest -m <marker>` over tests/models, tests/quantization, and tests/pipelines/testing_utils/quantization.py, giving the model-level mixin tests a nightly home with the backend dependencies installed. The torchao job additionally installs mslk. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
1 parent 1e4b353 commit 06bb7ff

1 file changed

Lines changed: 12 additions & 8 deletions

File tree

.github/workflows/nightly_tests.yml

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -343,19 +343,19 @@ jobs:
343343
matrix:
344344
config:
345345
- backend: "bitsandbytes"
346-
test_location: "bnb"
346+
marker: "bitsandbytes"
347347
additional_deps: ["peft"]
348348
- backend: "gguf"
349-
test_location: "gguf"
349+
marker: "gguf"
350350
additional_deps: ["peft", "kernels"]
351351
- backend: "torchao"
352-
test_location: "torchao"
353-
additional_deps: []
352+
marker: "torchao"
353+
additional_deps: ["mslk"]
354354
- backend: "optimum_quanto"
355-
test_location: "quanto"
355+
marker: "quanto"
356356
additional_deps: []
357357
- backend: "nvidia_modelopt"
358-
test_location: "modelopt"
358+
marker: "modelopt"
359359
additional_deps: []
360360
runs-on:
361361
group: aws-g6e-xlarge-plus
@@ -390,9 +390,12 @@ jobs:
390390
BIG_GPU_MEMORY: 40
391391
run: |
392392
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
393+
-m "${{ matrix.config.marker }}" \
393394
--make-reports=tests_${{ matrix.config.backend }}_torch_cuda \
394395
--report-log=tests_${{ matrix.config.backend }}_torch_cuda.log \
395-
tests/quantization/${{ matrix.config.test_location }}
396+
tests/models \
397+
tests/quantization \
398+
tests/pipelines/testing_utils/quantization.py
396399
- name: Failure short reports
397400
if: ${{ failure() }}
398401
run: |
@@ -440,9 +443,10 @@ jobs:
440443
BIG_GPU_MEMORY: 40
441444
run: |
442445
pytest -n 1 --max-worker-restart=0 --dist=loadfile \
446+
-k "TestPipelineQuantization" \
443447
--make-reports=tests_pipeline_level_quant_torch_cuda \
444448
--report-log=tests_pipeline_level_quant_torch_cuda.log \
445-
tests/quantization/test_pipeline_level_quantization.py
449+
tests/pipelines/testing_utils/quantization.py
446450
- name: Failure short reports
447451
if: ${{ failure() }}
448452
run: |

0 commit comments

Comments
 (0)