pytest-dist round robin to gpus

[drisspg]

Stacked PRs:

• [Ai-assisted] CLC work stealing #2218
• Nicer headdim error message #2227
• ->pytest-dist round robin to gpus #2241

---

pytest-dist round robin to gpus

Summary

With this conftest the pytest-xdist will launch on all available gpus:

Sorry neighbors :)

To only launch on 1 w/ pytest-xdist:
Just set the vis device before hand
CUDA_VISIBLE_DEVICES=0 pytest -n 12 tests/cute/test_flash_attn.py

[chatgpt-codex-connector]

You have reached your Codex usage limits for code reviews. You can see your limits in the Codex usage dashboard.
To continue using code reviews, you can upgrade your account or add credits to your account and enable them for code reviews in your settings.