JuliaGPU/CUDA.jl

Add `@non_differentiable` rules for basic random functions

Open

#2,153 创建于 2023年11月6日

在 GitHub 查看
 (2 评论) (0 反应) (0 负责人)Julia (274 fork)batch import
enhancementgood first issue

仓库指标

Star
 (1,408 star)
PR 合并指标
 (平均合并 5天 5小时) (30 天内合并 16 个 PR)

描述

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

贡献者指南

研究方向
从 ChainRules.jl 的 Random/random.jl 中适配 CUDA 的随机函数（如 zeros、rand、randn）的 @non differentiable 规则。
技术栈
julia
领域
backend
议题类型
功能
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
1
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1 小时以内
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
新近可参与
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
Julia
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
90