File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change 29
29
gpu-arch-type : " cuda"
30
30
gpu-arch-version : " 12.6"
31
31
- name : H100
32
- runs-on : linux.aws.h100
32
+ runs-on : linux.aws.h100.4
33
33
torch-spec : ' --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126'
34
34
gpu-arch-type : " cuda"
35
35
gpu-arch-version : " 12.4"
53
53
uv pip install -r dev-requirements.txt
54
54
uv pip install vllm
55
55
pip install .
56
- pytest test/float8 --verbose -s
57
- pytest test/integration --verbose -s
58
- pytest test/dtypes/test_affine_quantized_float.py --verbose -s
56
+ GPU_COUNT=$(nvidia-smi -L 2>/dev/null | wc -l)
57
+ if [ "$GPU_COUNT" -ge 4 ]; then
58
+ echo "Found $GPU_COUNT GPUs - running test_everything.sh"
59
+ ./test/float8/test_everything.sh
60
+ else
61
+ echo "Only $GPU_COUNT GPUs available. Need at least 4 GPUs to run test_everything.sh"
62
+ exit 1
63
+ fi
You can’t perform that action at this time.
0 commit comments