Skip to content

Commit 80fd973

Browse files
Revert "Undo cuda aware mpi"
This reverts commit a8d71b2.
1 parent aceeb16 commit 80fd973

File tree

1 file changed

+0
-13
lines changed

1 file changed

+0
-13
lines changed

source/source_pw/module_pwdft/operator_pw/op_exx_pw.cpp

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -244,26 +244,13 @@ void OperatorEXXPW<T, Device>::act_op(const int nbands,
244244

245245
// kpar: to real and bcast between same rank_in_pool
246246
// auto request = MPI_REQUEST_
247-
T* psi_mq_real_cpu = new T[wfcpw->nrxx];
248247
if (iq_pool == GlobalV::MY_POOL)
249248
{
250249
const T* psi_mq = get_pw(m_iband, iq_loc);
251250
wfcpw->recip_to_real(ctx, psi_mq, psi_mq_real, iq_loc);
252-
cudaMemcpy(psi_mq_real_cpu, psi_mq_real, wfcpw->nrxx, cudaMemcpyDeviceToHost);
253251
// send
254252
}
255253
MPI_Bcast(psi_mq_real, wfcpw->nrxx, MPI_DOUBLE_COMPLEX, iq_pool, KP_WORLD);
256-
if (iq_pool == GlobalV::MY_POOL)
257-
{
258-
const T* psi_mq = get_pw(m_iband, iq_loc);
259-
wfcpw->recip_to_real(ctx, psi_mq, psi_mq_real, iq_loc);
260-
// send
261-
}
262-
else
263-
{
264-
cudaMemcpy(psi_mq_real, psi_mq_real_cpu, wfcpw->nrxx, cudaMemcpyHostToDevice);
265-
}
266-
delete[] psi_mq_real_cpu;
267254
// std::cout << "psi_mq_real[0]: " << psi_mq_real[0] << std::endl;
268255
// if (GlobalV::RANK_IN_POOL == 1)
269256
// {

0 commit comments

Comments
 (0)