ensure fast thread local lookups occur once per access on macos

3 files changed, 26 insertions, 1 deletions

diff --git a/src/libstd/sys/redox/fast_thread_local.rs b/src/libstd/sys/redox/fast_thread_local.rs
index 67b92d490b2..c34cd575db7 100644
--- a/src/libstd/sys/redox/fast_thread_local.rs
+++ b/src/libstd/sys/redox/fast_thread_local.rs
@@ -1,4 +1,8 @@
 #![cfg(target_thread_local)]
 #![unstable(feature = "thread_local_internals", issue = "0")]
 
-pub use crate::sys_common::thread_local::register_dtor_fallback as register_dtor;
\ No newline at end of file
+pub use crate::sys_common::thread_local::register_dtor_fallback as register_dtor;
+
+pub unsafe fn lookup_once<T>(ptr: *const &T) -> &T {
+    *ptr
+}
diff --git a/src/libstd/sys/unix/fast_thread_local.rs b/src/libstd/sys/unix/fast_thread_local.rs
index c34c2e6e786..28fb9800541 100644
--- a/src/libstd/sys/unix/fast_thread_local.rs
+++ b/src/libstd/sys/unix/fast_thread_local.rs
@@ -82,3 +82,20 @@ pub unsafe fn register_dtor(t: *mut u8, dtor: unsafe extern fn(*mut u8)) {
         }
     }
 }
+
+#[cfg(not(target_os = "macos"))]
+pub unsafe fn lookup_once<T>(ptr: *const &T) -> &T {
+    *ptr
+}
+
+#[cfg(target_os = "macos")]
+pub unsafe fn lookup_once<T>(ptr: *const &T) -> &T {
+    // On macos, thread_local lookups can result in terrible code due to
+    // aggressive rerunning of the macos equivalent of `__tls_get_addr` - four
+    // lookups per actual reference in user code.
+    //
+    // Using a read_volatile on a value holding fast Key's address tricks the
+    // optimizer into only calling the macos get_addr equivalent once per time
+    // requested by the user.
+    crate::ptr::read_volatile(ptr)
+}
diff --git a/src/libstd/sys/windows/fast_thread_local.rs b/src/libstd/sys/windows/fast_thread_local.rs
index 31d0bd1e72e..566d5524fd7 100644
--- a/src/libstd/sys/windows/fast_thread_local.rs
+++ b/src/libstd/sys/windows/fast_thread_local.rs
@@ -2,3 +2,7 @@
 #![cfg(target_thread_local)]
 
 pub use crate::sys_common::thread_local::register_dtor_fallback as register_dtor;
+
+pub unsafe fn lookup_once<T>(ptr: *const &T) -> &T {
+    *ptr
+}