From 0d282ca4605ecaefd8f5b7d47fa9f4f61b513833 Mon Sep 17 00:00:00 2001
From: Marcin Kirsz <mkirsz@ed.ac.uk>
Date: Fri, 4 Oct 2024 12:37:35 +0100
Subject: [PATCH] Update
---
trainer.h | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/trainer.h b/trainer.h
index 60cd27a..92286b2 100644
--- a/trainer.h
+++ b/trainer.h
@@ -69,6 +69,7 @@ extern "C" void pdgemv_(char* transa, int* m, int* n, double* alpha, double* a,
class MPI_Trainer: public Trainer {
public:
+ int status;
int worker;
int tag;
int context1, context2;
@@ -92,8 +93,9 @@ class MPI_Trainer: public Trainer {
size_t phi_row = 0; // next row to be filled in the local phi array
MPI_Datatype rowvec, rowvecs;
- MPI_Trainer(Config &c):
- Trainer(c)
+ MPI_Trainer(Config &c, int &status):
+ Trainer(c),
+ status(status)
{}
void init(int &rank, int &ncpu) {
--
GitLab