Skip to content
Snippets Groups Projects
Commit 409a1fae authored by Marcin Kirsz's avatar Marcin Kirsz
Browse files

fix for solver when N > M

parent 899ca559
No related branches found
No related tags found
No related merge requests found
Pipeline #43488 passed
......@@ -222,9 +222,9 @@ class MPI_Trainer: public Trainer {
int lda2 = phi_rows2 > phi_cols2 ? phi_rows2 : phi_cols2;
descinit_( descPHI, &PHI_rows, &PHI_cols, &rnb1, &cnb1, &izero,
&izero, &context1, /*leading dimension*/&izero, &info);
&izero, &context1, /*leading dimension*/&phi_rows1, &info);
descinit_( descPHI2, &PHI_rows, &PHI_cols, &rnb2, &cnb2, &izero,
&izero, &context2, /*leading dimension*/&izero, &info2);
&izero, &context2, /*leading dimension*/&phi_cols1, &info2);
if(info != 0) {
printf("Error in descinit 1a, info = %d\n", info);
......@@ -235,9 +235,10 @@ class MPI_Trainer: public Trainer {
}
descinit_( descB, &PHI_cols, &ione, &rnb1, &cnb1, &izero,
int temp = PHI_rows > PHI_cols ? PHI_rows : PHI_cols;
descinit_( descB, &temp, &ione, &rnb1, &cnb1, &izero,
&izero, &context1, /*leading dimension*/&lda1, &info);
descinit_( descB2, &PHI_cols, &ione, &rnb2, &cnb2, &izero,
descinit_( descB2, &temp, &ione, &rnb2, &cnb2, &izero,
&izero, &context2, /*leading dimension*/&lda2, &info2);
if(info != 0) {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment