diff --git a/trainer.h b/trainer.h index 70956fdd966085f06775e7232011e57f40ec24d7..035d15225e43aa279eb255fb7a82ec5c22be4877 100644 --- a/trainer.h +++ b/trainer.h @@ -246,6 +246,8 @@ class MPI_Trainer: public Trainer { descinit_( sol, &PHI_cols, &ione, &rnb1, &cnb1, &izero, &izero, &context1, /*leading dimension*/&phi_cols1, &info3); + descinit_( sol2, &PHI_cols, &ione, &rnb2, &cnb2, &izero, + &izero, &context2, /*leading dimension*/&phi_cols2, &info3); if(info != 0) { printf("Error in descinit 1b, info = %d\n", info); @@ -293,7 +295,7 @@ class MPI_Trainer: public Trainer { descPHI2, dm2.T.ptr(), &ib, &jb, descB2, work2, &lwork2, &info2); // get weight vector, for context1 - pdgemr2d_(&PHI_cols, &ione, dm2.T.ptr(), &ione, &ione, descB2, + pdgemr2d_(&PHI_cols, &ione, dm2.T.ptr(), &ione, &ione, sol2, dm.T.ptr(), &ione, &ione, sol, &context1); if (rank==0) {