Commit 06a9409
ci: align integration test runners with downstream CIs, add multi-GPU peft
Key changes after digging into each downstream project's own CI:
Runner updates:
- transformers: T4 → A10G (bandb-aws-g5-4xlarge-plus). Current upstream
transformers quantization CI runs on g5.4xlarge (A10G); our earlier T4
choice came from a stale Feb-2024 fork.
- peft (single GPU): A10 → L4 (bandb-aws-g6-4xlarge-plus). Matches peft's
aws-g6-4xlarge-plus runner group exactly.
PEFT filter:
- Switched from `-m "single_gpu_tests and bitsandbytes"` (both test files)
to Benjamin Bossan's recommendation:
`-m single_gpu_tests -k PeftBnbGPUExampleTests tests/test_gpu_examples.py`.
Narrower scope (20 vs 86 tests) focused on the end-to-end QLoRA-style
integration signal, less noise from tests where bnb is incidental.
New multi-GPU peft job:
- Uses bandb-aws-g6-12xlarge-plus (4× L4, CUDA_VISIBLE_DEVICES=0,1) —
mirroring the legacy peft nightly-bnb.yml deleted in peft#2858.
- Filter: `-m multi_gpu_tests -k PeftBnbGPUExampleTests`.
- Note: this runner is being provisioned by infra; job will fail to pick
up a runner until that's done.
Accelerate:
- Added `-rs` to surface skip reasons. Previous run showed 26 silent skips
that produced a false "pass"; -rs will print the reason for each.
Report job's `needs:` updated to include test-peft-multigpu.
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>1 parent 2479b41 commit 06a9409
1 file changed
Lines changed: 67 additions & 8 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
43 | 43 | | |
44 | 44 | | |
45 | 45 | | |
46 | | - | |
| 46 | + | |
47 | 47 | | |
48 | | - | |
| 48 | + | |
49 | 49 | | |
50 | 50 | | |
51 | 51 | | |
| |||
140 | 140 | | |
141 | 141 | | |
142 | 142 | | |
143 | | - | |
| 143 | + | |
144 | 144 | | |
145 | 145 | | |
146 | 146 | | |
| |||
155 | 155 | | |
156 | 156 | | |
157 | 157 | | |
158 | | - | |
| 158 | + | |
159 | 159 | | |
160 | | - | |
| 160 | + | |
161 | 161 | | |
162 | 162 | | |
163 | 163 | | |
| |||
196 | 196 | | |
197 | 197 | | |
198 | 198 | | |
199 | | - | |
200 | | - | |
| 199 | + | |
| 200 | + | |
| 201 | + | |
201 | 202 | | |
202 | 203 | | |
203 | 204 | | |
| |||
211 | 212 | | |
212 | 213 | | |
213 | 214 | | |
| 215 | + | |
| 216 | + | |
| 217 | + | |
| 218 | + | |
| 219 | + | |
| 220 | + | |
| 221 | + | |
| 222 | + | |
| 223 | + | |
| 224 | + | |
| 225 | + | |
| 226 | + | |
| 227 | + | |
| 228 | + | |
| 229 | + | |
| 230 | + | |
| 231 | + | |
| 232 | + | |
| 233 | + | |
| 234 | + | |
| 235 | + | |
| 236 | + | |
| 237 | + | |
| 238 | + | |
| 239 | + | |
| 240 | + | |
| 241 | + | |
| 242 | + | |
| 243 | + | |
| 244 | + | |
| 245 | + | |
| 246 | + | |
| 247 | + | |
| 248 | + | |
| 249 | + | |
| 250 | + | |
| 251 | + | |
| 252 | + | |
| 253 | + | |
| 254 | + | |
| 255 | + | |
| 256 | + | |
| 257 | + | |
| 258 | + | |
| 259 | + | |
| 260 | + | |
| 261 | + | |
| 262 | + | |
| 263 | + | |
| 264 | + | |
| 265 | + | |
| 266 | + | |
| 267 | + | |
| 268 | + | |
| 269 | + | |
| 270 | + | |
| 271 | + | |
| 272 | + | |
214 | 273 | | |
215 | 274 | | |
216 | 275 | | |
| |||
221 | 280 | | |
222 | 281 | | |
223 | 282 | | |
224 | | - | |
| 283 | + | |
225 | 284 | | |
226 | 285 | | |
227 | 286 | | |
| |||
0 commit comments