From 053a6e1f39e18ebfd5908f46a6fea015137474b4 Mon Sep 17 00:00:00 2001
From: Marcin Kirsz <mkirsz@ed.ac.uk>
Date: Tue, 8 Oct 2024 14:40:25 +0100
Subject: [PATCH] fix for solver when N > M
---
trainer.h | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/trainer.h b/trainer.h
index 4163421..4a238b9 100644
--- a/trainer.h
+++ b/trainer.h
@@ -239,17 +239,16 @@ class MPI_Trainer: public Trainer {
}
- int temp = PHI_rows > PHI_cols ? PHI_rows : PHI_cols;
+ //int temp = PHI_rows > PHI_cols ? PHI_rows : PHI_cols;
descinit_( descB, &PHI_rows, &ione, &rnb1, &cnb1, &izero,
&izero, &context1, /*leading dimension*/&phi_rows1, &info);
descinit_( descB2, &PHI_rows, &ione, &rnb2, &cnb2, &izero,
&izero, &context2, /*leading dimension*/&phi_rows2, &info2);
descinit_( sol, &PHI_cols, &ione, &rnb1, &cnb1, &izero,
- &izero, &context1, /*leading dimension*/&phi_rows1, &info3);
-
+ &izero, &context1, /*leading dimension*/&phi_cols1, &info3);
descinit_( sol2, &PHI_cols, &ione, &rnb2, &cnb2, &izero,
- &izero, &context2, /*leading dimension*/&phi_rows2, &info4);
+ &izero, &context2, /*leading dimension*/&phi_cols2, &info4);
std::cout << "phi_rows1, phi_cols1 " << phi_rows1 << ", " << phi_cols1 << std::endl;
std::cout << "phi_rows2, phi_cols2 " << phi_rows2 << ", " << phi_cols2 << std::endl;
--
GitLab