Skip to content

Commit

Permalink
Revert changes to sync data structures
Browse files Browse the repository at this point in the history
There isn't currently a good reviewer for these, and I don't want to
remove things that will just be added again. I plan to make a separate
PR for these changes so the rest of the cleanup can land.
  • Loading branch information
jyn514 committed Mar 29, 2021
1 parent f352354 commit 526bb10
Showing 1 changed file with 110 additions and 0 deletions.
110 changes: 110 additions & 0 deletions compiler/rustc_data_structures/src/sync.rs
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,46 @@ cfg_if! {
use std::ops::Add;
use std::panic::{resume_unwind, catch_unwind, AssertUnwindSafe};

/// This is a single threaded variant of AtomicCell provided by crossbeam.
/// Unlike `Atomic` this is intended for all `Copy` types,
/// but it lacks the explicit ordering arguments.
#[derive(Debug)]
pub struct AtomicCell<T: Copy>(Cell<T>);

impl<T: Copy> AtomicCell<T> {
#[inline]
pub fn new(v: T) -> Self {
AtomicCell(Cell::new(v))
}

#[inline]
pub fn get_mut(&mut self) -> &mut T {
self.0.get_mut()
}
}

impl<T: Copy> AtomicCell<T> {
#[inline]
pub fn into_inner(self) -> T {
self.0.into_inner()
}

#[inline]
pub fn load(&self) -> T {
self.0.get()
}

#[inline]
pub fn store(&self, val: T) {
self.0.set(val)
}

#[inline]
pub fn swap(&self, val: T) -> T {
self.0.replace(val)
}
}

/// This is a single threaded variant of `AtomicU64`, `AtomicUsize`, etc.
/// It differs from `AtomicCell` in that it has explicit ordering arguments
/// and is only intended for use with the native atomic types.
Expand All @@ -59,6 +99,11 @@ cfg_if! {
}

impl<T: Copy> Atomic<T> {
#[inline]
pub fn into_inner(self) -> T {
self.0.into_inner()
}

#[inline]
pub fn load(&self, _: Ordering) -> T {
self.0.get()
Expand All @@ -68,6 +113,11 @@ cfg_if! {
pub fn store(&self, val: T, _: Ordering) {
self.0.set(val)
}

#[inline]
pub fn swap(&self, val: T, _: Ordering) -> T {
self.0.replace(val)
}
}

impl<T: Copy + PartialEq> Atomic<T> {
Expand Down Expand Up @@ -109,6 +159,22 @@ cfg_if! {
(oper_a(), oper_b())
}

pub struct SerialScope;

impl SerialScope {
pub fn spawn<F>(&self, f: F)
where F: FnOnce(&SerialScope)
{
f(self)
}
}

pub fn scope<F, R>(f: F) -> R
where F: FnOnce(&SerialScope) -> R
{
f(&SerialScope)
}

#[macro_export]
macro_rules! parallel {
($($blocks:tt),*) => {
Expand Down Expand Up @@ -180,6 +246,12 @@ cfg_if! {
pub fn new<F: FnMut(usize) -> T>(mut f: F) -> WorkerLocal<T> {
WorkerLocal(OneThread::new(f(0)))
}

/// Returns the worker-local value for each thread
#[inline]
pub fn into_inner(self) -> Vec<T> {
vec![OneThread::into_inner(self.0)]
}
}

impl<T> Deref for WorkerLocal<T> {
Expand Down Expand Up @@ -207,6 +279,16 @@ cfg_if! {
self.0
}

#[inline(always)]
pub fn get_mut(&mut self) -> &mut T {
&mut self.0
}

#[inline(always)]
pub fn lock(&self) -> &T {
&self.0
}

#[inline(always)]
pub fn lock_mut(&mut self) -> &mut T {
&mut self.0
Expand Down Expand Up @@ -236,6 +318,8 @@ cfg_if! {

pub use std::sync::atomic::{AtomicBool, AtomicUsize, AtomicU32, AtomicU64};

pub use crossbeam_utils::atomic::AtomicCell;

pub use std::sync::Arc as Lrc;
pub use std::sync::Weak as Weak;

Expand Down Expand Up @@ -437,6 +521,16 @@ impl<T> RwLock<T> {
RwLock(InnerRwLock::new(inner))
}

#[inline(always)]
pub fn into_inner(self) -> T {
self.0.into_inner()
}

#[inline(always)]
pub fn get_mut(&mut self) -> &mut T {
self.0.get_mut()
}

#[cfg(not(parallel_compiler))]
#[inline(always)]
pub fn read(&self) -> ReadGuard<'_, T> {
Expand All @@ -453,6 +547,11 @@ impl<T> RwLock<T> {
}
}

#[inline(always)]
pub fn with_read_lock<F: FnOnce(&T) -> R, R>(&self, f: F) -> R {
f(&*self.read())
}

#[cfg(not(parallel_compiler))]
#[inline(always)]
pub fn try_write(&self) -> Result<WriteGuard<'_, T>, ()> {
Expand Down Expand Up @@ -481,6 +580,11 @@ impl<T> RwLock<T> {
}
}

#[inline(always)]
pub fn with_write_lock<F: FnOnce(&mut T) -> R, R>(&self, f: F) -> R {
f(&mut *self.write())
}

#[inline(always)]
pub fn borrow(&self) -> ReadGuard<'_, T> {
self.read()
Expand Down Expand Up @@ -529,6 +633,12 @@ impl<T> OneThread<T> {
inner,
}
}

#[inline(always)]
pub fn into_inner(value: Self) -> T {
value.check();
value.inner
}
}

impl<T> Deref for OneThread<T> {
Expand Down

0 comments on commit 526bb10

Please sign in to comment.