FluxML/Zygote.jl

Unable to differentiate a broadcasted constructor with CUDA array inputs

Aberta

#1.528 aberto em 3 de out. de 2024

Ver no GitHub
 (2 comentários) (0 reação) (0 responsável)Julia (220 forks)batch import
CUDAhelp wanted

Métricas do repositório

Stars
 (1.568 estrelas)
Métricas de merge de PR
 (Mesclagem média 2d 16h) (39 fundiu PRs em 30d)

Description

Due to a CUDA bug: https://github.com/JuliaGPU/CUDA.jl/issues/2514, crossposted for reference.

Encountered when differentiating over:

using Distributions
using CUDA
a = cu(ones(5)); b = cu(zeros(5));

Normal.(a, b)

Guia do colaborador

Direção de pesquisa
O problema está bloqueado por um bug do CUDA.jl (referência: JuliaGPU/CUDA.jl#2514). Para prosseguir, primeiro entenda esse bug e suas implicações na diferenciação de construtores transmitidos (broadcast) no Zygote. Em seguida, examine o tratamento de chamadas transmitidas no Zygote e veja se uma solução alternativa é possível. O exemplo fornecido usa Distributions.Normal com arrays CUDA; revise a interação entre Distributions e Zygote. Ainda não há PRs vinculados ou responsáveis atribuídos.
Pilha de tecnologia
None
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.
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.
Precisa de investigação
Prerequisites
Julia
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
30