Skip to content

Commit 15ac6cd

Browse files
committed
Remember to throw errors instead of returning them
1 parent 72be1db commit 15ac6cd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/layers/normalise.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ end
4646

4747
dropout_mask(rng::CUDA.RNG, x::CuArray, p; kwargs...) = _dropout_mask(rng, x, p; kwargs...)
4848
dropout_mask(rng, x::CuArray, p; kwargs...) =
49-
ArgumentError("x isa CuArray, but rng isa $(typeof(rng)). dropout_mask only support CUDA.RNG for CuArrays.")
49+
throw(ArgumentError("x isa CuArray, but rng isa $(typeof(rng)). dropout_mask only support CUDA.RNG for CuArrays."))
5050
dropout_mask(rng, x, p; kwargs...) = _dropout_mask(rng, x, p; kwargs...)
5151
function _dropout_mask(rng, x, p; dims=:)
5252
y = rand!(rng, similar(x, _dropout_shape(x, dims)))

0 commit comments

Comments
 (0)