JuliaGPU/CUDA.jl
Auf GitHub ansehenAdd `@non_differentiable` rules for basic random functions
Open
#2.153 geöffnet am 6. Nov. 2023
enhancementgood first issue
Repository-Metriken
- Stars
- (1.408 Stars)
- PR-Merge-Metriken
- (Durchschn. Merge 5T 5h) (16 gemergte PRs in 30 T)
Beschreibung
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