JuliaGPU/CUDA.jl

Add `@non_differentiable` rules for basic random functions

Open

#2153 aperta il 6 nov 2023

Vedi su GitHub
 (2 commenti) (0 reazioni) (0 assegnatari)Julia (274 fork)batch import
enhancementgood first issue

Metriche repository

Star
 (1408 star)
Metriche merge PR
 (Merge medio 5g 5h) (16 PR mergiate in 30 g)

Descrizione

I need to have these lines in my codes to make AD works.

@non_differentiable CUDA.zeros(::Any...)
@non_differentiable CUDA.rand(::Any...)
@non_differentiable CUDA.randn(::Any...)

Can we add them to the extension? Adapting from https://github.com/JuliaDiff/ChainRules.jl/blob/main/src/rulesets/Random/random.jl

Guida contributor

Direzione di ricerca
Adatta le regole @non differentiable da Random/random.jl di ChainRules.jl per le funzioni casuali di CUDA come zeros, rand, randn.
Tech stack
julia
Dominio
backend
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
1
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Meno di 1 ora
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
Julia
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
90