JuliaGPU/CUDA.jl

Add `@non_differentiable` rules for basic random functions

Open

#2.153 aberto em 6 de nov. de 2023

Ver no GitHub
 (2 comments) (0 reactions) (0 assignees)Julia (274 forks)batch import
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

Guia do colaborador