simon/moonpool
1
0
Fork 0
mirror of https://github.com/c-cube/moonpool.git synced 2025-12-06 03:05:30 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

perf ws_deque: use bitmasks instead of modulo op

Browse source
This commit is contained in:
Simon Cruanes 2023-10-25 09:57:05 -04:00
parent 78407c495d
commit 3d7e272d01
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/ws_deque_.ml
View file

@ -30,12 +30,13 @@ end = struct
{ log_size; arr = Array.make (1 lsl log_size) None } { log_size; arr = Array.make (1 lsl log_size) None }
let[@inline] get (self : _ t) (i : int) : 'a = let[@inline] get (self : _ t) (i : int) : 'a =
match Array.unsafe_get self.arr (i mod size self) with match Array.unsafe_get self.arr (i land ((1 lsl self.log_size) - 1)) with
| Some x -> x | Some x -> x
| None -> assert false | None -> assert false
let[@inline] set (self : 'a t) (i : int) (x : 'a) : unit = let[@inline] set (self : 'a t) (i : int) (x : 'a) : unit =
Array.unsafe_set self.arr (i mod size self) (Some x) assert (i >= 0);
Array.unsafe_set self.arr (i land ((1 lsl self.log_size) - 1)) (Some x)
let grow (self : _ t) ~bottom ~top : 'a t = let grow (self : _ t) ~bottom ~top : 'a t =
let new_arr = create ~log_size:(self.log_size + 1) () in let new_arr = create ~log_size:(self.log_size + 1) () in
@ -73,6 +74,7 @@ let push (self : 'a t) (x : 'a) : unit =
Only if it seems too big do we actually read [t]. *) Only if it seems too big do we actually read [t]. *)
let size_approx = b - t_approx in let size_approx = b - t_approx in
if size_approx >= CA.size self.arr - 1 then ( if size_approx >= CA.size self.arr - 1 then (
(* we need to read the actual value of [top], which might entail contention. *)
let t = A.get self.top in let t = A.get self.top in
self.top_cached <- t; self.top_cached <- t;
let size = b - t in let size = b - t in