remove duplicate allowscalar call

2019-02-11 20:32:23 +05:30 · 2019-02-11 20:32:23 +05:30 · d16ef75b1c
parent 1ada9afe81
commit d16ef75b1c
1 changed files with 1 additions and 1 deletions
--- a/test/cuda/cuda.jl
+++ b/test/cuda/cuda.jl
@ -39,9 +39,9 @@ Flux.back!(sum(l))
 end

@testset "onecold gpu" begin
-  CuArrays.allowscalar(false)
  y = Flux.onehotbatch(ones(3), 1:10) |> gpu;
  @test Flux.onecold(y) isa CuArray
+  @test y[:,:] isa Flux.OneHotMatrix{<:CuArray}
  @test y[3,:] isa CuArray
 end