#0: Scheduling fix

tenstorrent · Oct 29, 2024 · ed11135 · ed11135
1 parent b85798a
commit ed11135
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tt_metal/impl/buffers/buffer.cpp b/tt_metal/impl/buffers/buffer.cpp
@@ -241,7 +241,6 @@ std::shared_ptr<Buffer> Buffer::create(
     }
 
     buffer->device_->push_work([buffer] {
-        AllocationStatus next_status = AllocationStatus::ALLOCATED;
         try {
             buffer->address_ = detail::AllocateBuffer(buffer.get());
         } catch(...) {
@@ -299,8 +298,7 @@ void Buffer::deleter(Buffer* buffer) {
 }
 
 void Buffer::deallocate_impl() {
-    auto status = allocation_status_.load(std::memory_order::relaxed);
-    if (status == AllocationStatus::DEALLOCATED || status == AllocationStatus::ALLOCATION_FAILED) {
+    if (allocation_status_.load(std::memory_order::relaxed) != AllocationStatus::ALLOCATED) {
         return;
     }
 
@@ -330,6 +328,10 @@ uint32_t Buffer::address() const {
         return address_;
     }
 
+    if (device_->can_use_passthrough_scheduling()) {
+        return address_;
+    }
+
     std::unique_lock lock(allocation_mutex_);
     allocation_cv_.wait(lock, [this] { return this->allocation_status_.load(std::memory_order::relaxed) != AllocationStatus::ALLOCATION_REQUESTED; });
     return address_;