#0: update tensor hash

tenstorrent · Jan 20, 2024 · 95760b4 · 95760b4
1 parent aa58e30
commit 95760b4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tt_eager/tensor/tensor.hpp b/tt_eager/tensor/tensor.hpp
@@ -131,7 +131,7 @@ Tensor create_sharded_device_tensor(const Shape& shape, DataType data_type, Layo
 
 namespace std {
 template <>
-struct hash<Tensor> {
+struct hash<tt::tt_metal::Tensor> {
     std::size_t operator()(const Tensor &tensor) const {
         std::size_t hash = std::hash<TensorMemoryLayout>()(tensor.memory_config().memory_layout);
         hash ^= std::hash<BufferType>()(tensor.memory_config().buffer_type);

diff --git a/tt_eager/tensor/types.hpp b/tt_eager/tensor/types.hpp
@@ -286,7 +286,7 @@ bool operator!=(const ShardSpec& spec_a, const ShardSpec& spec_b);
 
 namespace std {
 template <>
-struct hash<MemoryConfig> {
+struct hash<tt::tt_metal::MemoryConfig> {
     std::size_t operator()(const MemoryConfig &mem_config) const {
         std::size_t hash = std::hash<TensorMemoryLayout>()(mem_config.memory_layout);
         hash ^= std::hash<BufferType>()(mem_config.buffer_type);