Update runtime code to to use ::ttnn::SmallVector<int64_t> for ::ttnn…

…::permute() - Fixes compile error "no known conversion from 'std::vector<long>' to 'const SmallVector<int64_t>'"
tenstorrent · Jan 4, 2025 · 67aedca · 67aedca
1 parent edc6bd2
commit 67aedca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/runtime/lib/ttnn/operations/data_movement/permute.cpp b/runtime/lib/ttnn/operations/data_movement/permute.cpp
@@ -16,8 +16,8 @@ void run(const ::tt::target::ttnn::PermuteOp *op, ProgramContext &context) {
   const ::ttnn::Tensor &in = tensorPool.at(op->in()->global_id());
   DEBUG_ASSERT(in.is_allocated());
 
-  std::vector<int64_t> permutation(op->permutation()->begin(),
-                                   op->permutation()->end());
+  ::ttnn::SmallVector<int64_t> permutation(op->permutation()->begin(),
+                                           op->permutation()->end());
   std::optional<tt::tt_metal::MemoryConfig> memoryConfig =
       op->memory_config() ? std::make_optional(utils::createMemoryConfig(
                                 op->memory_config(), op->out()))