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