FluxML/Zygote.jl
在 GitHub 查看Unable to differentiate a broadcasted constructor with CUDA array inputs
Open
#1,528 建立於 2024年10月3日
CUDAhelp wanted
倉庫指標
- Star
- (1,568 star)
- PR 合併指標
- (平均合併 2天 16小時) (30 天內合併 39 個 PR)
描述
Due to a CUDA bug: https://github.com/JuliaGPU/CUDA.jl/issues/2514, crossposted for reference.
Encountered when differentiating over:
using Distributions
using CUDA
a = cu(ones(5)); b = cu(zeros(5));
Normal.(a, b)