diff --git a/lib/Conversion/TTNNToEmitC/TTNNToEmitC.cpp b/lib/Conversion/TTNNToEmitC/TTNNToEmitC.cpp index a22d1e4c6..56a363d38 100644 --- a/lib/Conversion/TTNNToEmitC/TTNNToEmitC.cpp +++ b/lib/Conversion/TTNNToEmitC/TTNNToEmitC.cpp @@ -568,11 +568,11 @@ class RepeatOpConversionPattern ConversionPatternRewriter &rewriter) const override { ttnn::ShapeAttr repeatDims = repeatOp.getRepeatDimsAttr(); - // Create ttnn::Shape() call + // Create ttnn::SimpleShape() call // emitc::CallOpaqueOp shapeOp = ttnn_to_emitc::utils::createShapeOp( rewriter, repeatDims, repeatOp.getLoc(), - ttnn_to_emitc::utils::ShapeType::Shape); + ttnn_to_emitc::utils::ShapeType::SimpleShape); // Create operands vector // diff --git a/runtime/lib/ttnn/operations/data_movement/repeat.cpp b/runtime/lib/ttnn/operations/data_movement/repeat.cpp index 39bae64df..2fd66992e 100644 --- a/runtime/lib/ttnn/operations/data_movement/repeat.cpp +++ b/runtime/lib/ttnn/operations/data_movement/repeat.cpp @@ -13,7 +13,7 @@ void run(const ::tt::target::ttnn::RepeatOp *op, ProgramContext &context) { DEBUG_ASSERT(in.is_allocated()); const auto *fbShape = op->repeat_dims(); const std::vector repeatDims(fbShape->begin(), fbShape->end()); - ::ttnn::Shape repeatDimsShape(repeatDims); + ::ttnn::SimpleShape repeatDimsShape(repeatDims); ::ttnn::Tensor out = ::ttnn::repeat(in, repeatDimsShape); tensorPool.insert_or_assign(op->out()->global_id(), out); } diff --git a/third_party/CMakeLists.txt b/third_party/CMakeLists.txt index f8d05f169..e7f5fcf35 100644 --- a/third_party/CMakeLists.txt +++ b/third_party/CMakeLists.txt @@ -1,6 +1,6 @@ include(ExternalProject) -set(TT_METAL_VERSION "064646adfcb5f0b923a24a633d532552bd43e06d") +set(TT_METAL_VERSION "9d69fb143bac50983dff914c5348539d0a7d2021") if ("$ENV{ARCH_NAME}" STREQUAL "grayskull") set(ARCH_NAME "grayskull")