Remember to throw errors instead of returning them

FluxML · Jan 27, 2022 · 15ac6cd · 15ac6cd
1 parent 72be1db
commit 15ac6cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layers/normalise.jl b/src/layers/normalise.jl
@@ -46,7 +46,7 @@ end
 
 dropout_mask(rng::CUDA.RNG, x::CuArray, p; kwargs...) = _dropout_mask(rng, x, p; kwargs...)
 dropout_mask(rng, x::CuArray, p; kwargs...) =
-  ArgumentError("x isa CuArray, but rng isa $(typeof(rng)). dropout_mask only support CUDA.RNG for CuArrays.")
+  throw(ArgumentError("x isa CuArray, but rng isa $(typeof(rng)). dropout_mask only support CUDA.RNG for CuArrays."))
 dropout_mask(rng, x, p; kwargs...) = _dropout_mask(rng, x, p; kwargs...)
 function _dropout_mask(rng, x, p; dims=:)
   y = rand!(rng, similar(x, _dropout_shape(x, dims)))