create IO subscriber bitset for tracking done

1 files changed, 6 insertions, 0 deletions

diff --git a/lib/block_allocator.cpp b/lib/block_allocator.cpp
index b2da9ea..ad39159 100644
--- a/lib/block_allocator.cpp
+++ b/lib/block_allocator.cpp
@@ -74,6 +74,12 @@ void BlockActor::handle_top_alloc(const TopAllocMessage &, const Theron::Address
 {
     MESSAGE_TRACER();
 
+    if (this->block_state == BLOCK_STATE_DONE)
+    {
+        this->Send(0, from); //ACK
+        return;
+    }
+
     //allocate output buffers which will also wake up the task
     const size_t num_outputs = this->get_num_outputs();
     this->output_buffer_tokens.resize(num_outputs);