try fix pytest

deepmodeling · Nov 7, 2024 · b6ceba0 · b6ceba0
1 parent ff71d0d
commit b6ceba0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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}