pymc-devs/pymc

Only sample using a single thread when sampling multiple chains in parallel

Chiusa

#5453 aperta il 9 feb 2022

Vedi su GitHubGuida contributor
 (1 commento) (0 reazioni) (0 assegnatari)Python (1902 fork)batch import
help wantedrequest discussionv4

Metriche repository

Star
 (7926 stelle)
Metriche merge PR
 (Merge medio 11g) (7 PR mergiate in 30 g)

Descrizione

My prior of what cores=4 does when calling pm.sample() is that pymc would use 4 of my cores. However, because of multi-threading this might not be the case. Furthermore, multi-threading is causing problems when running multiple chains in parallel: https://github.com/pymc-devs/pymc/issues/3140

I thus think we should set os.environ['OMP_NUM_THREADS'] = '1' if we're sampling on > 1 core.

Guida contributor

Direzione di ricerca
Imposta os.environ['OMP NUM THREADS'] = '1' in pm.sample() quando si campionano più catene per evitare conflitti di multi threading. Controlla l'issue #3140 per il contesto. Assicurati che il campionamento a catena singola non sia influenzato.
Tech stack
python
Dominio
backend
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
Pythonpymc
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
70