JuliaGPU/CUDA.jl
Ver no GitHubAdd `@non_differentiable` rules for basic random functions
Open
#2.153 aberto em 6 de nov. de 2023
enhancementgood first issue
Métricas do repositório
- Stars
- (1.408 stars)
- Métricas de merge de PR
- (Mesclagem média 5d 5h) (16 fundiu PRs em 30d)
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