Skip to content

Commit 5656842

Browse files
authored
Some tweaks
1 parent 34404f2 commit 5656842

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/UnweightedSamplingMulti.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ end
9999
if s.seen_k <= n
100100
@inbounds s.value[s.seen_k] = el
101101
if s.seen_k === n
102-
s = recompute_skip!(s, n)
102+
s = @inline recompute_skip!(s, n)
103103
new_values = sample(s.rng, s.value, n, ordered=is_ordered(s))
104104
@inbounds for i in 1:n
105105
s.value[i] = new_values[i]
@@ -115,7 +115,7 @@ end
115115
s.value[r], s.value[j] = s.value[j], el
116116
update_order_multi!(s, r, j)
117117
end
118-
s = recompute_skip!(s, n)
118+
s = @inline recompute_skip!(s, n)
119119
end
120120
return s
121121
end
@@ -151,7 +151,7 @@ end
151151

152152
function recompute_skip!(s::SampleMultiAlgL, n)
153153
@update s.state += randexp(s.rng)
154-
@update s.skip_k = s.seen_k-ceil(Int, randexp(s.rng)/log(1-exp(-s.state/n)))
154+
@update s.skip_k = s.seen_k-ceil(Int, randexp(s.rng)/log1p(-exp(-s.state/n)))
155155
return s
156156
end
157157
function recompute_skip!(s::SampleMultiAlgRSWRSKIP, n)

0 commit comments

Comments
 (0)