From be66ef5c2f4ddb8d85d6ffd6f9ce68eef061ba94 Mon Sep 17 00:00:00 2001
From: Marcin Kirsz <mkirsz@ed.ac.uk>
Date: Tue, 8 Oct 2024 11:12:58 +0100
Subject: [PATCH] Changed inheritace, fix
---
trainer.h | 11 ++---------
1 file changed, 2 insertions(+), 9 deletions(-)
diff --git a/trainer.h b/trainer.h
index 9856314..50108d1 100644
--- a/trainer.h
+++ b/trainer.h
@@ -294,14 +294,11 @@ class MPI_Trainer: public Trainer {
class MPI_Trainer_HOST: public MPI_Trainer {
private:
- int &rank;
- int &ncpu;
std::vector<std::tuple<std::string,int,int>> wpckgs;
public:
MPI_Trainer_HOST(int &rank, int &ncpu):
- rank(rank),
- ncpu(ncpu)
+ MPI_Trainer(rank. ncpu),
{}
void prep_wpckgs() {
@@ -427,14 +424,10 @@ class MPI_Trainer_HOST: public MPI_Trainer {
};
class MPI_Trainer_WORKER: public MPI_Trainer {
- private:
- int &rank;
- int &ncpu;
public:
MPI_Trainer_WORKER(int &rank, int &ncpu):
- rank(rank),
- ncpu(ncpu)
+ MPI_Trainer(rank. ncpu),
{}
bool release_tag() {
--
GitLab