Report MPS and ROCm devices in `ofl doctor`

Chiusa
  • Qualcosa non va?
#12 1 commento 0 reazioni 0 assegnatari Vedi su GitHub

I maintainer di solito rispondono entro 1 giorno

Una pull request collegata è già stata integrata.

  • #29 di @eldertek — integrata

Valutazione

Difficoltà
3/5
Tempo stimato
1-2 giorni
Idoneità per principianti
25/100
Tipo di issue
Funzionalità
Chiarezza
Specificata chiaramente
Stato di attività
Ferma
Stack tecnologico
python, pytorch
Ambito
devtools, machine-learning

Direzione di ricerca

Start with src/ofl/doctor.py and the existing tests in tests/test_doctor.py; stub the torch calls to follow the CUDA branch. Add reporting for MPS and ROCm, including device name and available memory, and clear absent-backend output. Update ARCHITECTURE §1 “Today” and check that make doctor, make check, and make test meet the listed criteria. An open pull request (#29) is already linked.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Descrizione

agent-authored area/python good first issue help wanted layer/1-hardware type/feature

Context

ofl doctor reports CUDA GPUs only. The swarm is meant to include Apple Silicon and AMD machines (ARCHITECTURE §1, next milestones), and RFC 0001 Rung 1 needs a Mac.

Task

Extend src/ofl/doctor.py so it detects and reports Apple MPS (torch.backends.mps.is_available()) and ROCm (torch.version.hip) devices alongside CUDA: device name, memory where torch exposes it, and backend.

Definition of done

  • make doctor shows an MPS or ROCm line when present and says plainly when absent
  • Unit tests in tests/test_doctor.py cover the CUDA, MPS, ROCm and CPU-only branches with torch calls stubbed
  • ARCHITECTURE §1 "Today" paragraph updated
  • make check and make test pass
  • Commits signed off (git commit -s)

Pointers

Skills: Python

Agent: omp / claude-opus-5-5 (issue drafted by an agent; a maintainer reviews the PR)

Lingua principale
Python
Stelle
1
Fork
2
Merge medio
2h 4m
PR unite (30g)
23

Preparare l'ambiente

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Altre issue di OpenFrontierLabs/openfrontier

Tutte le issue di OpenFrontierLabs/openfrontier

Issue simili

Altre issue su Python

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.