Fix MultinomialSampler overflow

JuliaStats · Jun 30, 2023 · e17caa0 · e17caa0
1 parent 25b3149
commit e17caa0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/samplers/multinomial.jl b/src/samplers/multinomial.jl
@@ -52,7 +52,7 @@ function _rand!(rng::AbstractRNG, s::MultinomialSampler,
                 x::AbstractVector{<:Real})
     n = s.n
     k = length(s)
-    if n^2 > k
+    if n > sqrt(k)
         multinom_rand!(rng, n, s.prob, x)
     else
         # Use an alias table