Uplift third_party/tt-metal to b552fb8ae14236777983e3521a312dd78432f0…

…e1 2025-02-07 (#2139)

This PR uplifts the third_party/tt-metal to the
b552fb8ae14236777983e3521a312dd78432f0e1

- Update include locations for small_vector after metal commit b552fb8ae1

---------

Co-authored-by: kmabeeTT <118925087+kmabeeTT@users.noreply.github.com>
Co-authored-by: brataTT <achoudhury@tenstorrent.com>

lib/OpModel/TTNN/Conversion.cpp

@@ -39,7 +39,7 @@ getDataType(const mlir::tt::ttnn::TTNNLayoutAttr layout) {
 }
 
 ::ttnn::Shape getShape(const ::llvm::ArrayRef<int64_t> shape) {
-  ::tt::tt_metal::SmallVector<uint32_t> small_vector_shape;
+  ::tt::stl::SmallVector<uint32_t> small_vector_shape;
   for (const auto &dim : shape) {
     small_vector_shape.push_back(static_cast<uint32_t>(dim));
   }

lib/OpModel/TTNN/MetalHeaders.h

@@ -56,6 +56,7 @@
 #include "tt-metalium/core_coord.hpp"
 #include "tt-metalium/device_impl.hpp"
 #include "tt-metalium/host_api.hpp"
+#include "tt-metalium/small_vector.hpp"
 #include "ttnn/graph/graph_processor.hpp"
 #include "ttnn/graph/graph_query_op_constraints.hpp"
 #include "ttnn/graph/graph_query_op_runtime.hpp"
@@ -64,7 +65,6 @@
 #include "ttnn/operations/eltwise/unary/unary.hpp"
 #include "ttnn/operations/matmul/matmul.hpp"
 #include "ttnn/operations/normalization/softmax/softmax.hpp"
-#include "ttnn/tensor/shape/small_vector.hpp"
 #include "ttnn/tensor/tensor.hpp"
 #include "ttnn/tensor/tensor_spec.hpp"
 #include "ttnn/tensor/types.hpp"

runtime/lib/ttnn/runtime.cpp

@@ -2,6 +2,7 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+#include "tt-metalium/small_vector.hpp"
 #include "tt/runtime/detail/common.h"
 #include "tt/runtime/detail/debug.h"
 #include "tt/runtime/detail/logger.h"
@@ -12,7 +13,6 @@
 #include "tt/runtime/utils.h"
 #include "ttmlir/Target/TTNN/Target.h"
 #include "ttmlir/Version.h"
-#include "ttnn/tensor/shape/small_vector.hpp"
 #include "ttnn/tensor/types.hpp"
 
 namespace tt::runtime::ttnn {

third_party/CMakeLists.txt

@@ -1,6 +1,6 @@
 include(ExternalProject)
 
-set(TT_METAL_VERSION "b093594aa5fc01f7c008a0b5ffddbcf2713bd8f7")
+set(TT_METAL_VERSION "b552fb8ae14236777983e3521a312dd78432f0e1")
 
 if ("$ENV{ARCH_NAME}" STREQUAL "grayskull")
   set(ARCH_NAME "grayskull")

tools/ttnn-standalone/ttnn-precompiled.hpp

@@ -23,10 +23,10 @@
 #include "operations/moreh/moreh_cumsum/moreh_cumsum.hpp"
 #include "operations/normalization/softmax/softmax.hpp"
 #include "operations/reduction/generic/generic_reductions.hpp"
-#include "tensor/shape/small_vector.hpp"
 #include "tensor/tensor.hpp"
 #include "tensor/types.hpp"
 #include "tt-metalium/bfloat16.hpp"
+#include "tt-metalium/small_vector.hpp"
 #include "types.hpp"
 // ANCHOR_END: standalone_includes