Skip to content

Commit

Permalink
Remove push_unchecked
Browse files Browse the repository at this point in the history
On recent Rust compilers, it produces the same assembly as `Vec::push`, so
there is no benefit in keeping the function around, besides that it avoids
occasionally nerd-sniping people in trying to understand Rust's UB and how
to write the function correctly.

Signed-off-by: Moritz Hoffmann <[email protected]>
  • Loading branch information
antiguru committed Nov 9, 2023
1 parent 07b987a commit 9e34fb9
Showing 1 changed file with 3 additions and 12 deletions.
15 changes: 3 additions & 12 deletions src/trace/implementations/merge_batcher.rs
Original file line number Diff line number Diff line change
Expand Up @@ -121,15 +121,6 @@ where
}
}


#[inline]
unsafe fn push_unchecked<T>(vec: &mut Vec<T>, element: T) {
debug_assert!(vec.len() < vec.capacity());
let idx = vec.len();
vec.set_len(idx + 1);
::std::ptr::write(vec.get_unchecked_mut(idx), element);
}

pub struct MergeSorter<D: Ord, T: Ord, R: Semigroup> {
queue: Vec<Vec<Vec<(D, T, R)>>>, // each power-of-two length list of allocations.
stash: Vec<Vec<(D, T, R)>>,
Expand Down Expand Up @@ -242,14 +233,14 @@ impl<D: Ord, T: Ord, R: Semigroup> MergeSorter<D, T, R> {
(&x.0, &x.1).cmp(&(&y.0, &y.1))
};
match cmp {
Ordering::Less => { unsafe { push_unchecked(&mut result, head1.pop_front().unwrap()); } }
Ordering::Greater => { unsafe { push_unchecked(&mut result, head2.pop_front().unwrap()); } }
Ordering::Less => result.push(head1.pop_front().unwrap()),
Ordering::Greater => result.push(head2.pop_front().unwrap()),
Ordering::Equal => {
let (data1, time1, mut diff1) = head1.pop_front().unwrap();
let (_data2, _time2, diff2) = head2.pop_front().unwrap();
diff1.plus_equals(&diff2);
if !diff1.is_zero() {
unsafe { push_unchecked(&mut result, (data1, time1, diff1)); }
result.push((data1, time1, diff1));
}
}
}
Expand Down

0 comments on commit 9e34fb9

Please sign in to comment.