Merge #926

926: Fix binarycrossentropy on CuArrays r=MikeInnes a=janEbert Ref #889. Co-authored-by: janEbert <janpublicebert@posteo.net>
FluxML · Nov 8, 2019 · 4f42fb2 · 4f42fb2
2 parents 9d05afa + 3dceef4
commit 4f42fb2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl
@@ -1,3 +1,4 @@
+using CuArrays
 using NNlib: logsoftmax, logσ
 
 # Cost functions
@@ -35,6 +36,9 @@ Return `-y*log(ŷ + ϵ) - (1-y)*log(1-ŷ + ϵ)`. The ϵ term provides numerica
 """
 binarycrossentropy(ŷ, y; ϵ=eps(ŷ)) = -y*log(ŷ + ϵ) - (1 - y)*log(1 - ŷ + ϵ)
 
+# Re-definition to fix interaction with CuArrays.
+CuArrays.@cufunc binarycrossentropy(ŷ, y; ϵ=eps(ŷ)) = -y*log(ŷ + ϵ) - (1 - y)*log(1 - ŷ + ϵ)
+
 """
     logitbinarycrossentropy(logŷ, y)