Use volatile TLS in library/cpp/yt

1 files changed, 9 insertions, 5 deletions

diff --git a/library/cpp/yt/cpu_clock/clock.cpp b/library/cpp/yt/cpu_clock/clock.cpp
index eda3f03bbc..0ed65e3b92 100644
--- a/library/cpp/yt/cpu_clock/clock.cpp
+++ b/library/cpp/yt/cpu_clock/clock.cpp
@@ -4,6 +4,8 @@
 
 #include <library/cpp/yt/assert/assert.h>
 
+#include <library/cpp/yt/misc/tls.h>
+
 namespace NYT {
 
 ////////////////////////////////////////////////////////////////////////////////
@@ -31,14 +33,16 @@ double GetTicksToMicroseconds()
 
 TCalibrationState GetCalibrationState(TCpuInstant cpuInstant)
 {
-    thread_local TCalibrationState State;
+    YT_THREAD_LOCAL(TCalibrationState) State;
+
+    auto& state = GetTlsRef(State);
 
-    if (State.CpuInstant + CalibrationCpuPeriod < cpuInstant) {
-        State.CpuInstant = cpuInstant;
-        State.Instant = TInstant::Now();
+    if (state.CpuInstant + CalibrationCpuPeriod < cpuInstant) {
+        state.CpuInstant = cpuInstant;
+        state.Instant = TInstant::Now();
     }
 
-    return State;
+    return state;
 }
 
 TCalibrationState GetCalibrationState()