fix pair access

tenstorrent · Mar 3, 2025 · 8a9e4b1 · 8a9e4b1
1 parent 92f391b
commit 8a9e4b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ttnn/cpp/ttnn/tensor/storage.hpp b/ttnn/cpp/ttnn/tensor/storage.hpp
@@ -330,7 +330,7 @@ struct MultiDeviceStorage {
         TT_ASSERT(
             buffer_it->device() == device,
             "Mismatch between device derived from buffer and device derived from MultiDeviceStorage.");
-        return *buffer_it;
+        return buffer_it->second;
     }
 
     inline std::shared_ptr<Buffer>& get_buffer_for_device(IDevice* device) {