JuliaGPU/CUDA.jl

Add `@non_differentiable` rules for basic random functions

Open

#2 153 ouverte le 6 nov. 2023

Voir sur GitHub
 (2 commentaires) (0 réactions) (0 assignés)Julia (274 forks)batch import
enhancementgood first issue

Métriques du dépôt

Stars
 (1 408 stars)
Métriques de merge PR
 (Merge moyen 5j 5h) (16 PRs mergées en 30 j)

Description

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

Guide contributeur