JuliaGPU/CUDA.jl
GitHub で見るAdd `@non_differentiable` rules for basic random functions
Open
#2,153 opened on 2023年11月6日
enhancementgood first issue
Repository metrics
- Stars
- (1,408 stars)
- PR merge metrics
- (平均マージ 5d 5h) (30d で 16 merged PRs)
説明
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