Skip to content

Commit c2761af

Browse files
committed
slightly update Lanczos to make it identical to v1.3.1
1 parent 4573cdc commit c2761af

2 files changed

Lines changed: 6 additions & 0 deletions

File tree

ChASE-MPI/impl/chase_mpidla.hpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -897,6 +897,7 @@ class ChaseMpiDLA : public ChaseMpiDLAInterface<T>
897897
nvtxRangePop();
898898
nvtxRangePushA("allreduce");
899899
#endif
900+
900901
AllReduce(allreduce_backend, rsd + locked, unconverged,
901902
getMPI_Type<Base<T>>(), MPI_SUM, row_comm_, mpi_wrapper_);
902903
// Base<T> *resid_h;
@@ -1334,6 +1335,7 @@ class ChaseMpiDLA : public ChaseMpiDLAInterface<T>
13341335
nvtxRangePop();
13351336
#endif
13361337
*/
1338+
std::cout << C[0] << " " << ritzVc[0] << std::endl;
13371339
dla_->LanczosDos(idx, m, ritzVc);
13381340
}
13391341

@@ -1415,6 +1417,9 @@ class ChaseMpiDLA : public ChaseMpiDLAInterface<T>
14151417
std::swap(v1_, v0_);
14161418
std::swap(v1_, v2_);
14171419
}
1420+
1421+
Memcpy(memcpy_mode[2], C, v1_, m_ * sizeof(T));
1422+
14181423
#ifdef USE_NSIGHT
14191424
nvtxRangePop();
14201425
#endif

ChASE-MPI/impl/chase_mpidla_blaslapack.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,7 @@ class ChaseMpiDLABlaslapack : public ChaseMpiDLAInterface<T>
310310
t_gemm(CblasColMajor, CblasNoTrans, CblasNoTrans, m_, idx, m, &alpha,
311311
C_, m_, ritzVc, m, &beta, C2_, m_);
312312
std::memcpy(C_, C2_, m * m_ * sizeof(T));
313+
313314
}
314315
void Lanczos(std::size_t M, int idx, Base<T>* d, Base<T>* e,
315316
Base<T>* r_beta) override

0 commit comments

Comments
 (0)