From 02daac6e9c0736063863e5596bb7fa4f0be1b32b Mon Sep 17 00:00:00 2001 From: Vladimir Milosevic <157983820+vmilosevic@users.noreply.github.com> Date: Wed, 4 Dec 2024 04:55:08 +0100 Subject: [PATCH] Uplift third_party/tt-metal to 6a524ab817aeb09c273e37254f39ad8124ddf2f8 2024-12-03 (#1463) * Uplift third_party/tt-metal to 6a524ab817aeb09c273e37254f39ad8124ddf2f8 2024-12-03 * reflect tt-metal refactor to uplift tt-metal - fixed build error due to refactor in https://github.com/tenstorrent/tt-metal/commit/3584ac9512fef17e96f7e838bff1fd12c97798fb --------- Co-authored-by: kmitrovicTT <169657397+kmitrovicTT@users.noreply.github.com> Co-authored-by: Brata Choudhury --- runtime/include/tt/runtime/detail/ttnn.h | 2 +- runtime/lib/ttnn/operations/pool/maxpool2d.cpp | 6 ++++-- third_party/CMakeLists.txt | 2 +- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/runtime/include/tt/runtime/detail/ttnn.h b/runtime/include/tt/runtime/detail/ttnn.h index e7b8fbcf2..31b979e13 100644 --- a/runtime/include/tt/runtime/detail/ttnn.h +++ b/runtime/include/tt/runtime/detail/ttnn.h @@ -25,7 +25,7 @@ #include "ttnn/operations/embedding/embedding.hpp" #include "ttnn/operations/matmul/matmul.hpp" #include "ttnn/operations/normalization/softmax/softmax.hpp" -#include "ttnn/operations/pool/maxpool/max_pool2d.hpp" +#include "ttnn/operations/pool/generic/generic_pools.hpp" #include "ttnn/operations/reduction/generic/generic_reductions.hpp" #include "ttnn/tensor/host_buffer/functions.hpp" #include "ttnn/tensor/tensor.hpp" diff --git a/runtime/lib/ttnn/operations/pool/maxpool2d.cpp b/runtime/lib/ttnn/operations/pool/maxpool2d.cpp index c405a86f1..ddbe639c7 100644 --- a/runtime/lib/ttnn/operations/pool/maxpool2d.cpp +++ b/runtime/lib/ttnn/operations/pool/maxpool2d.cpp @@ -47,8 +47,10 @@ preshardForMaxPool2d(const ::tt::target::ttnn::MaxPool2dOp *op, void run(const ::tt::target::ttnn::MaxPool2dOp *op, ProgramContext &context) { ProgramTensorPool &tensorPool = context.getTensorPool(); - const ::ttnn::operations::pool::MaxPool2DOp operation = - ::ttnn::operations::pool::MaxPool2DOp(); + const ::ttnn::operations::pool::Pool2DOp< + ::ttnn::operations::pool::Pool2DType::MAX_POOL2D> + operation = ::ttnn::operations::pool::Pool2DOp< + ::ttnn::operations::pool::Pool2DType::MAX_POOL2D>(); ::ttnn::Tensor input = tensorPool.at(op->in()->global_id()); DEBUG_ASSERT(input.is_allocated()); diff --git a/third_party/CMakeLists.txt b/third_party/CMakeLists.txt index bf28aebc9..4c814fdc8 100644 --- a/third_party/CMakeLists.txt +++ b/third_party/CMakeLists.txt @@ -1,6 +1,6 @@ include(ExternalProject) -set(TT_METAL_VERSION "ab3dc0c4f5c3ce9722261c878970bfa92a212fc9") +set(TT_METAL_VERSION "6a524ab817aeb09c273e37254f39ad8124ddf2f8") if ("$ENV{ARCH_NAME}" STREQUAL "grayskull") set(ARCH_NAME "grayskull")