Skip to content

Commit 93d32b1

Browse files
committed
change op
1 parent 00d5f31 commit 93d32b1

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

source/module_psi/psi_init.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ void PSIInit<T, Device>::initialize_psi(Psi<std::complex<double>>* psi,
139139
this->psi_initer->init_psig(psi_cpu->get_pointer(), ik);
140140
if (psi_device->get_pointer() != psi_cpu->get_pointer())
141141
{
142-
castmem_h2d_op()(ctx, cpu_ctx, psi_device->get_pointer(), psi_cpu->get_pointer(), nbands_start * nbasis);
142+
syncmem_h2d_op()(ctx, cpu_ctx, psi_device->get_pointer(), psi_cpu->get_pointer(), nbands_start * nbasis);
143143
}
144144

145145
std::vector<typename GetTypeReal<T>::type> etatom(nbands_start, 0.0);

source/module_psi/psi_init.h

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,8 +82,7 @@ class PSIInit
8282

8383
//-------------------------OP--------------------------------------------
8484
using syncmem_complex_op = base_device::memory::synchronize_memory_op<T, Device, Device>;
85-
using castmem_h2d_op
86-
= base_device::memory::cast_memory_op<T, T, Device, base_device::DEVICE_CPU>;
85+
using syncmem_h2d_op = base_device::memory::synchronize_memory_op<T, Device, base_device::DEVICE_CPU>;
8786
};
8887

8988
///@brief allocate the wavefunction

0 commit comments

Comments
 (0)