#0: Build fix

tenstorrent · Oct 23, 2024 · f954e36 · f954e36
1 parent a47056b
commit f954e36
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/ttnn/cpp/ttnn/tensor/tensor.cpp b/ttnn/cpp/ttnn/tensor/tensor.cpp
@@ -339,8 +339,8 @@ void Tensor::deallocate(bool force) {
                                 });
 
                             for (auto worker : this->workers) {
-                                worker->push_work(std::make_shared<std::function<void()>>(
-                                    [worker, dealloc_lambda]() mutable { (*dealloc_lambda)(worker); }));
+                                worker->push_work(
+                                    [worker, dealloc_lambda]() mutable { (*dealloc_lambda)(worker); });
                             }
                         }
                     } else {

diff --git a/ttnn/cpp/ttnn/tensor/tensor_ops.cpp b/ttnn/cpp/ttnn/tensor/tensor_ops.cpp
@@ -26,9 +26,8 @@
 namespace{
     inline void SynchronizeWorkerThreads(const std::vector<Device*>& workers) {
         // Push empty work to threads and ensure its been picked up
-        static auto empty_work = std::make_shared<std::function<void()>>([](){});
         for (auto target_device : workers) {
-            target_device->work_executor.push_work(empty_work);
+            target_device->work_executor.push_work([](){});
         }
         // Block until work has been picked up, to flush the queue
         for (auto target_device : workers) {