pymc-devs/pymc

CheckParameterValues not replaced in inner graphs

Aberta

#6.697 aberto em 28 de abr. de 2023

Ver no GitHubGuia do colaborador
 (3 comentários) (0 reação) (0 responsável)Python (1.902 forks)batch import
help wantedlogprob

Métricas do repositório

Stars
 (7.926 estrelas)
Métricas de merge de PR
 (Mesclagem média 11d) (7 fundiu PRs em 30d)

Description

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

Guia do colaborador

Direção de pesquisa
O problema é que a reescrita CheckParameterValues não é aplicada a grafos internos (por exemplo, dentro de Scan ou OpFromGraph não inline). O código relevante está nas linhas 1142-1144 de pymc/pytensorf.py. A correção envolveria garantir que essa reescrita seja aplicada recursivamente a todos os subgrafos. Comece examinando como as reescritas são aplicadas no processo de compilação e observe o exemplo na issue #6696 para reproduzir a diferença de desempenho. Considere modificar a lógica de aplicação da reescrita para percorrer os grafos internos.
Pilha de tecnologia
python
Domain
backend
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
Pythonbasic PyMC knowledge
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
55