JuliaGPU/CUDA.jl
Voir sur GitHubAdd `@non_differentiable` rules for basic random functions
Open
#2 153 ouverte le 6 nov. 2023
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