Runtime - Update createBufferFromTensorRef() for Buffer API changes i…

…n tt-metal - Matches changes from tt-metal PR 14394
tenstorrent · Oct 30, 2024 · a2e64a2 · a2e64a2
1 parent caa7b90
commit a2e64a2
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/runtime/include/tt/runtime/detail/ttmetal.h b/runtime/include/tt/runtime/detail/ttmetal.h
@@ -159,12 +159,12 @@ createBufferFromTensorRef(::tt::tt_metal::Device *device,
                           .page_size = pageSize,
                           .buffer_type = bufferType,
                           .buffer_layout = TensorMemoryLayout::BLOCK_SHARDED,
-                          .shard_parameters = shardSpecBuffer,
-                          .allocate = false};
-  std::shared_ptr<::tt::tt_metal::Buffer> buffer =
-      ::tt::tt_metal::CreateBuffer(shardedBufferConfig);
+                          .shard_parameters = shardSpecBuffer};
+
   assert(tensorRef->address());
-  buffer->set_address(tensorRef->address());
+  std::shared_ptr<::tt::tt_metal::Buffer> buffer =
+      ::tt::tt_metal::CreateBuffer(shardedBufferConfig, tensorRef->address());
+
   return buffer;
 }
 #pragma clang diagnostic pop