TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow scalar indexing or onehotbatch tests will fail

This commit is contained in:
Manjunath Bhat 2019-07-12 17:56:47 +05:30 committed by GitHub
parent 33c8d84a60
commit 2b379d0ec0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/cuda/cuda.jl
View File

@ -6,8 +6,6 @@ using Zygote
@testset "CuArrays" begin
CuArrays.allowscalar(false)
x = param(randn(5, 5))
cx = gpu(x)
@test cx isa CuArray