bigscience-workshop/petals

Allocate all GPU memory on startup

Open

#308 aperta il 24 apr 2023

Vedi su GitHub
 (0 commenti) (0 reazioni) (0 assegnatari)Python (424 fork)batch import
help wanted

Metriche repository

Star
 (8248 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

Suggested by @slush0 on Discord.

It would be great to allocate all GPU memory on startup (including max space for attention cache and max space for intermediate tensors created during backward).

Right now, the memory is allocated on-demand, and people may see OOMs only after a few days of running the server. Allocating everything right away would allows to see OOMs right away and choose a correct number of blocks before leaving a server to run in the background.

Any other way to check if we have enough GPU memory on startup would be also good.

Guida contributor

Direzione di ricerca
Esamina l'attuale allocazione della memoria GPU su richiesta nel codice, identifica dove vengono allocati la cache di attenzione e i tensori intermedi durante il passaggio all'indietro e implementa un meccanismo per pre allocare la memoria massima richiesta all'avvio. Considera l'uso di pool di memoria CUDA o di pre allocazione di tensori PyTorch.
Tech stack
python
Dominio
backendperformance
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-2 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonGPU memory management
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
50