From b6ceba0fc268a7c27ff50b2f2e13dd9cb190c28c Mon Sep 17 00:00:00 2001
From: Qianruipku <Terry_Liu@pku.edu.cn>
Date: Thu, 7 Nov 2024 18:28:15 +0800
Subject: [PATCH] try fix pytest

---
 python/pyabacus/src/ModuleBase/CMakeLists.txt | 1 +
 python/pyabacus/src/ModuleNAO/CMakeLists.txt  | 1 +
 2 files changed, 2 insertions(+)

diff --git a/python/pyabacus/src/ModuleBase/CMakeLists.txt b/python/pyabacus/src/ModuleBase/CMakeLists.txt
index f150cf1b5e..7ce5fb5e3b 100644
--- a/python/pyabacus/src/ModuleBase/CMakeLists.txt
+++ b/python/pyabacus/src/ModuleBase/CMakeLists.txt
@@ -2,6 +2,7 @@ list(APPEND pymodule_base
     ${PROJECT_SOURCE_DIR}/src/ModuleBase/py_base_math.cpp
     ${BASE_PATH}/kernels/math_op.cpp
     ${BASE_PATH}/module_device/memory_op.cpp
+    ${BASE_PATH}/module_device/device.cpp
     )
 
 pybind11_add_module(_base_pack MODULE ${pymodule_base})
diff --git a/python/pyabacus/src/ModuleNAO/CMakeLists.txt b/python/pyabacus/src/ModuleNAO/CMakeLists.txt
index 65e209ca40..53600a08f3 100644
--- a/python/pyabacus/src/ModuleNAO/CMakeLists.txt
+++ b/python/pyabacus/src/ModuleNAO/CMakeLists.txt
@@ -17,6 +17,7 @@ list(APPEND _naos
     ${ABACUS_SOURCE_DIR}/module_base/kernels/math_op.cpp
     # ${ABACUS_SOURCE_DIR}/module_psi/kernels/psi_memory_op.cpp
     ${ABACUS_SOURCE_DIR}/module_base/module_device/memory_op.cpp
+    ${ABACUS_SOURCE_DIR}/module_base/module_device/device.cpp
 )
 add_library(naopack SHARED 
     ${_naos}