From 8121b00b523d51a3a90a94adeca411a98ee86326 Mon Sep 17 00:00:00 2001
From: Marcin Kirsz <mkirsz@ed.ac.uk>
Date: Fri, 4 Oct 2024 12:30:23 +0100
Subject: [PATCH] Update

---
 trainer.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trainer.h b/trainer.h
index 05457bd..69b65ba 100644
--- a/trainer.h
+++ b/trainer.h
@@ -291,7 +291,7 @@ class MPI_Trainer_HOST {
         MPI_Send (&rows_accepted, 1, MPI_INT, tr.worker, TadahCLI::DATA_TAG, MPI_COMM_WORLD);
       }
     }
-    void release() {
+    void release_tag(int &count) {
       int rows_available;
       MPI_Recv (&rows_available, 1, MPI_INT, tr.worker, TadahCLI::WORK_TAG, MPI_COMM_WORLD, &status);
       // there is no more work so release a worker if full
-- 
GitLab