Skip to content

Commit 14fe324

Browse files
committed
add syncmem_op for PyDiagoDavSubspace
1 parent a02c35b commit 14fe324

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

python/pyabacus/src/hsolver/py_diago_dav_subspace.hpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,10 @@ class PyDiagoDavSubspace
161161
int nband;
162162

163163
std::unique_ptr<hsolver::Diago_DavSubspace<std::complex<double>, base_device::DEVICE_CPU>> obj;
164+
165+
base_device::DEVICE_CPU* ctx = {};
166+
using syncmem_op = base_device::memory::
167+
synchronize_memory_op<std::complex<double>, base_device::DEVICE_CPU, base_device::DEVICE_CPU>;
164168
};
165169

166170
} // namespace py_hsolver

0 commit comments

Comments
 (0)