Replace Shape with SimpleShape after metal change 452da80

tenstorrent · Jan 31, 2025 · f0389ef · f0389ef
1 parent b159ebc
commit f0389ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<uint32_t> 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);
 }