diff --git a/trainer.h b/trainer.h index 3284f13821edc0d195ec8628fffc9e4d2c8ad8be..1416693b183f9aa78544529d80509a542e0b381e 100644 --- a/trainer.h +++ b/trainer.h @@ -266,6 +266,7 @@ class MPI_Trainer: public Trainer { double *b2 = dm2.T.ptr(); double wkopt2; int lwork2 = -1; // query -> get size of the work matrix + std::cout << "rank: " << rank << " 1. trans: " << trans << " PHI_cols " << PHI_cols << " PHI_rows " << PHI_rows << std::endl; pdgels_(&trans, &PHI_rows, &PHI_cols, &nrhs, dm2.Phi.ptr(), &ia, &ja, descPHI2, b2, &ib, &jb, descB2, &wkopt2, &lwork2, &info2); if (info2 != 0) { @@ -273,6 +274,7 @@ class MPI_Trainer: public Trainer { } lwork2 = (int)wkopt2; double *work2 = new double[lwork2]; + std::cout << "rank: " << rank << " 2. trans: " << trans << " PHI_cols " << PHI_cols << " PHI_rows " << PHI_rows << std::endl; pdgels_(&trans, &PHI_rows, &PHI_cols, &nrhs, dm2.Phi.ptr(), &ia, &ja, descPHI2, b2, &ib, &jb, descB2, work2, &lwork2, &info2);