Track location for atomic new/drop

src/rt/atomic.rs

@@ -313,6 +313,9 @@ impl<T: Numeric> Atomic<T> {
                     state.stores[index].value = T::into_u64(self.0);
 
                     if !std::thread::panicking() {
+                        state
+                            .unsync_mut_locations
+                            .track(location!(), &execution.threads);
                         state.track_unsync_mut(&execution.threads);
                     }
                 });
@@ -358,6 +361,7 @@ impl State {
         };
 
         // All subsequent accesses must happen-after.
+        state.unsync_mut_locations.track(location, &threads);
         state.track_unsync_mut(threads);
 
         // Store the initial thread

src/rt/mod.rs

@@ -1,3 +1,7 @@
+#[macro_use]
+mod location;
+pub(crate) use self::location::Location;
+
 mod access;
 use self::access::Access;
 
@@ -10,10 +14,6 @@ pub(crate) use self::arc::Arc;
 mod atomic;
 pub(crate) use self::atomic::{fence, Atomic};
 
-#[macro_use]
-mod location;
-pub(crate) use self::location::Location;
-
 mod cell;
 pub(crate) use self::cell::Cell;
 

tests/atomic.rs

@@ -6,6 +6,14 @@ use loom::thread;
 use std::sync::atomic::Ordering::{AcqRel, Acquire, Relaxed, Release};
 use std::sync::Arc;
 
+#[test]
+fn load_after_new() {
+    loom::model(|| {
+        let a = AtomicUsize::new(0);
+        a.load(Relaxed);
+    });
+}
+
 #[test]
 #[should_panic]
 fn invalid_unsync_load_relaxed() {