buffer unique id generator to atomic

tenstorrent · Nov 26, 2024 · 8d15695 · 8d15695
1 parent c8ce757
commit 8d15695
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tt_metal/impl/buffers/buffer.cpp b/tt_metal/impl/buffers/buffer.cpp
@@ -13,6 +13,7 @@
 #include "tt_metal/types.hpp"
 
 #include <algorithm>
+#include <atomic>
 #include <mutex>
 #include <utility>
 #include "tt_metal/common/base.hpp"
@@ -25,7 +26,7 @@ namespace tt {
 
 namespace tt_metal {
 
-size_t Buffer::next_unique_id = 0;
+std::atomic<size_t> Buffer::next_unique_id = 0;
 
 std::ostream& operator<<(std::ostream& os, const ShardSpec& spec) {
     tt::stl::reflection::operator<<(os, spec);
@@ -249,7 +250,7 @@ Buffer::Buffer(
     if (size != 0) {
         validate_buffer_size_and_page_size(size, page_size, buffer_type, buffer_layout, shard_parameters);
     }
-    unique_id_ = next_unique_id++;
+    unique_id_ = next_unique_id.fetch_add(1);
 }
 
 std::shared_ptr<Buffer> Buffer::create(

diff --git a/tt_metal/impl/buffers/buffer.hpp b/tt_metal/impl/buffers/buffer.hpp
@@ -303,7 +303,7 @@ class Buffer final {
 
     std::weak_ptr<Buffer> weak_self;
     size_t unique_id_;
-    static size_t next_unique_id;
+    static std::atomic<size_t> next_unique_id;
 };
 
 }  // namespace v0