From 3618448a194ea1a3c88a0a350e7ca5eb68ab275a Mon Sep 17 00:00:00 2001
From: haozhihan <haozhi.han@stu.pku.edu.cn>
Date: Mon, 9 Dec 2024 20:58:54 +0800
Subject: [PATCH] add & for vector

---
 python/pyabacus/src/hsolver/py_diago_dav_subspace.hpp | 4 ++--
 python/pyabacus/src/hsolver/py_diago_david.hpp        | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/pyabacus/src/hsolver/py_diago_dav_subspace.hpp b/python/pyabacus/src/hsolver/py_diago_dav_subspace.hpp
index f2865b3a78..112eb836e2 100644
--- a/python/pyabacus/src/hsolver/py_diago_dav_subspace.hpp
+++ b/python/pyabacus/src/hsolver/py_diago_dav_subspace.hpp
@@ -101,12 +101,12 @@ class PyDiagoDavSubspace
 
     int diag(
         std::function<py::array_t<std::complex<double>>(py::array_t<std::complex<double>>)> mm_op,
-        std::vector<double> precond_vec,
+        std::vector<double>& precond_vec,
         int dav_ndim,
         double tol,
         int max_iter,
         bool need_subspace,
-        std::vector<double> diag_ethr,
+        std::vector<double>& diag_ethr,
         bool scf_type,
         hsolver::diag_comm_info comm_info
     ) {
diff --git a/python/pyabacus/src/hsolver/py_diago_david.hpp b/python/pyabacus/src/hsolver/py_diago_david.hpp
index 02d621027e..8a8d2c727e 100644
--- a/python/pyabacus/src/hsolver/py_diago_david.hpp
+++ b/python/pyabacus/src/hsolver/py_diago_david.hpp
@@ -101,10 +101,10 @@ class PyDiagoDavid
 
     int diag(
         std::function<py::array_t<std::complex<double>>(py::array_t<std::complex<double>>)> mm_op,
-        std::vector<double> precond_vec,
+        std::vector<double>& precond_vec,
         int dav_ndim,
         double tol,
-        std::vector<double> diag_ethr,
+        std::vector<double>& diag_ethr,
         int max_iter,
         bool use_paw,
         hsolver::diag_comm_info comm_info