Skip to content

Commit 72be1db

Browse files
committed
Fix UndefVar on dropout GPU test
1 parent 5aa0b12 commit 72be1db

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

test/cuda/layers.jl

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

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

test/cuda/runtests.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
using Flux, Test, CUDA
22
using Zygote
33
using Zygote: pullback
4+
using Random
45

56
@info "Testing GPU Support"
67
CUDA.allowscalar(false)

0 commit comments

Comments
 (0)