Skip to content

Commit 3f07fe5

Browse files
committed
add ref link
1 parent dc67b9d commit 3f07fe5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/ext/CUDA/cuda.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ using Bijectors, CUDA, Distributions, Flux, LinearAlgebra, Test
88
@testset "rand with CUDA" begin
99

1010
# Bijectors versions use dot for broadcasting, which causes issues with CUDA.
11+
# https://github.com/TuringLang/Bijectors.jl/blob/6f0d383f73afd150a018b65a3ea4ac9306065d38/src/bijectors/planar_layer.jl#L65-L80
1112
function Bijectors.get_u_hat(u::CuVector{T}, w::CuVector{T}) where {T<:Real}
1213
wT_u = dot(w, u)
1314
scale = (Bijectors.LogExpFunctions.log1pexp(-wT_u) - 1) / sum(abs2, w)

0 commit comments

Comments
 (0)