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

cpu -> test_throws

This commit is contained in:
Dhairya Gandhi 2019-12-20 23:02:44 +05:30
parent efa2cbfd0e
commit b1e68813a8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/cuda/layers.jl
View File

@ -6,7 +6,7 @@
# generic movement tests
@test_broken gradient(x -> sum(gpu(x)), rand(3,3)) isa Tuple
@test gradient(x -> sum(cpu(x)), gpu(rand(3,3))) isa Tuple
@test_throws ErrorException gradient(x -> sum(cpu(x)), gpu(rand(3,3))) isa Tuple
function gradtest(layers, args...; name = "Conv", xs = rand(Float32, 28, 28, 1, 1))
@testset "$name GPU grad tests" begin