Allow scalar indexing or onehotbatch tests will fail
This commit is contained in:
parent
33c8d84a60
commit
2b379d0ec0
@ -6,8 +6,6 @@ using Zygote
|
|||||||
|
|
||||||
@testset "CuArrays" begin
|
@testset "CuArrays" begin
|
||||||
|
|
||||||
CuArrays.allowscalar(false)
|
|
||||||
|
|
||||||
x = param(randn(5, 5))
|
x = param(randn(5, 5))
|
||||||
cx = gpu(x)
|
cx = gpu(x)
|
||||||
@test cx isa CuArray
|
@test cx isa CuArray
|
||||||
|
Loading…
Reference in New Issue
Block a user