Don't invoke GPU crossentropy with integers.

Broadcasting log on integers does not work.
2020-01-31 08:22:54 +01:00 · 2020-01-31 08:22:54 +01:00 · e2c2ec5575
parent e66a7f130f
commit e2c2ec5575
1 changed files with 1 additions and 1 deletions
--- a/test/cuda/cuda.jl
+++ b/test/cuda/cuda.jl
@ -25,7 +25,7 @@ cm = gpu(m)
@test all(p isa CuArray for p in params(cm))
@test cm(gpu(rand(10, 10))) isa CuArray{Float32,2}

-x = [1,2,3]
+x = [1.,2.,3.]
 cx = gpu(x)
@test Flux.crossentropy(x,x) ≈ Flux.crossentropy(cx,cx)
@test Flux.crossentropy(x,x, weight=1.0) ≈ Flux.crossentropy(cx,cx, weight=1.0)