pymc-devs/pymc

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

Fechada

#5.453 aberto em 9 de fev. de 2022

Ver no GitHubGuia do colaborador
 (1 comentário) (0 reação) (0 responsável)Python (1.902 forks)batch import
help wantedrequest discussionv4

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

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.

Guia do colaborador

Direção de pesquisa
Defina os.environ['OMP NUM THREADS'] = '1' em pm.sample() ao amostrar múltiplas cadeias para evitar conflitos de multithreading. Consulte a issue #3140 para obter contexto. Certifique se de que a amostragem de cadeia única não seja afetada.
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.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
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
Pythonpymc
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
70