From 18d7cc99b6a56e9bb19253f4af3fbe071ec52a76 Mon Sep 17 00:00:00 2001
From: Istvan Haller <31476121+ihaller@users.noreply.github.com>
Date: Wed, 23 Jun 2021 10:50:28 +0100
Subject: [PATCH] Extended TrivialInitAtomic (#340)

---
 src/ds/helpers.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/ds/helpers.h b/src/ds/helpers.h
index 4ba019b..49211a3 100644
--- a/src/ds/helpers.h
+++ b/src/ds/helpers.h
@@ -219,6 +219,21 @@ namespace snmalloc
     {
       return this->ref().compare_exchange_strong(exp, des, mo);
     }
+
+    SNMALLOC_FAST_PATH T
+    exchange(T des, std::memory_order mo = std::memory_order_seq_cst) noexcept
+    {
+      return this->ref().exchange(des, mo);
+    }
+
+    template<typename Q>
+    SNMALLOC_FAST_PATH
+      typename std::enable_if<std::is_integral<Q>::value, Q>::type
+      fetch_add(
+        Q arg, std::memory_order mo = std::memory_order_seq_cst) noexcept
+    {
+      return this->ref().fetch_add(arg, mo);
+    }
   };
 
   static_assert(sizeof(TrivialInitAtomic<char>) == sizeof(char));