only run the remainder of the gpu test suite if one case passes first (#2009) [skip ci]

* only run the remainder of the gpu test suite if one case passes first

* also reduce the test matrix
This commit is contained in:
Wing Lian
2024-10-31 13:45:01 -04:00
committed by GitHub
parent 3591bcfaf9
commit 052a9a79b4
2 changed files with 41 additions and 20 deletions

View File

@@ -82,13 +82,6 @@ jobs:
num_gpus: 1
axolotl_extras: mamba-ssm
nightly_build: "true"
- cuda: 121
cuda_version: 12.1.1
python_version: "3.11"
pytorch: 2.3.1
num_gpus: 1
axolotl_extras: mamba-ssm
nightly_build: "true"
- cuda: 124
cuda_version: 12.4.1
python_version: "3.11"