Skip to content

Commit 6af147a

Browse files
committed
Test dropout(::MersenneTwister, ::CuArray, ...)
1 parent 57a5df9 commit 6af147a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/cuda/layers.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -282,6 +282,7 @@ end
282282
end
283283

284284
@testset "Dropout RNGs" begin
285+
@test_throws ArgumentError dropout(MersenneTwister(), CUDA.rand(Float32, 2, 3), 0.1)
285286
@testset for layer in (Dropout, AlphaDropout)
286287
m = layer(0.1; rng = MersenneTwister(123))
287288
@test_throws ErrorException gpu(m)

0 commit comments

Comments
 (0)