File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Expand file tree Collapse file tree 1 file changed +9
-1
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"
56
56
pytest test/float8 --verbose -s
57
57
pytest test/integration --verbose -s
58
58
pytest test/dtypes/test_affine_quantized_float.py --verbose -s
59
+ GPU_COUNT=$(nvidia-smi -L 2>/dev/null | wc -l)
60
+ if [ "$GPU_COUNT" -ge 4 ]; then
61
+ echo "Found $GPU_COUNT GPUs - running test_everything.sh"
62
+ ./test/float8/test_everything.sh
63
+ else
64
+ echo "Only $GPU_COUNT GPUs available. Need at least 4 GPUs to run test_everything.sh"
65
+ exit 0
66
+ fi
You can’t perform that action at this time.
0 commit comments