pymc-devs/pymc

CheckParameterValues not replaced in inner graphs

Open

#6697 aperta il 28 apr 2023

Vedi su GitHub
 (3 commenti) (0 reazioni) (0 assegnatari)Python (1902 fork)batch import
help wantedlogprob

Metriche repository

Star
 (7926 star)
Metriche merge PR
 (Merge medio 11g 6h) (12 PR mergiate in 30 g)

Descrizione

We include these rewrites when compiling logp graphs, but this won't replace inner logp graphs (e.g, inside Scan or not-inlined OpFromGraph)

https://github.com/pymc-devs/pymc/blob/371472dcf1c44f99a9ad4e7317606404b2a47a0e/pymc/pytensorf.py#L1142-L1144

The example in #6696 suffers from this. Sampling is quite faster when the rewrite is manually added

Guida contributor

Direzione di ricerca
Il problema è che la riscrittura CheckParameterValues non viene applicata ai grafi interni (ad esempio, all'interno di Scan o di OpFromGraph non in linea). Il codice rilevante si trova alle righe 1142-1144 di pymc/pytensorf.py. La correzione richiederebbe di garantire che questa riscrittura venga applicata ricorsivamente a tutti i sottografi. Inizia esaminando come le riscritture vengono applicate durante il processo di compilazione e guarda l'esempio nell'issue #6696 per riprodurre la differenza di prestazioni. Considera di modificare la logica di applicazione della riscrittura per attraversare i grafi interni.
Tech stack
python
Dominio
backend
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
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
Pythonbasic PyMC knowledge
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
55