FluxML/Zygote.jl

Unable to differentiate a broadcasted constructor with CUDA array inputs

Open

#1 528 ouverte le 3 oct. 2024

Voir sur GitHub
 (2 commentaires) (0 réactions) (0 assignés)Julia (220 forks)batch import
CUDAhelp wanted

Métriques du dépôt

Stars
 (1 568 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

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)

Guide contributeur

Direction de recherche
Le problème est bloqué par un bug de CUDA.jl (référence : JuliaGPU/CUDA.jl#2514). Pour avancer, comprenez d'abord ce bug et ses implications sur la différenciation des constructeurs diffusés (broadcast) dans Zygote. Ensuite, examinez la gestion des appels diffusés dans Zygote et voyez si un contournement est possible. L'exemple fourni utilise Distributions.Normal avec des tableaux CUDA ; examinez l'interaction entre Distributions et Zygote. Il n'existe pas encore de PR liés ou de personnes assignées.
Stack technique
Aucun
Domaine
backend
Type d'issue
Bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Investigation requise
Prérequis
Julia
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
30