From 9c191b65f53803bb605969aebc53c4bfa4dd6e56 Mon Sep 17 00:00:00 2001
From: Chen Nuo <49788094+Cstandardlib@users.noreply.github.com>
Date: Wed, 8 Jan 2025 22:37:49 +0800
Subject: [PATCH] Fix test dependency on DCU

---
 .../module_base/module_container/ATen/ops/test/CMakeLists.txt | 2 +-
 source/module_elecstate/test/CMakeLists.txt                   | 4 ++--
 source/module_elecstate/test/elecstate_base_test.cpp          | 2 ++
 source/module_elecstate/test/elecstate_print_test.cpp         | 2 ++
 4 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/source/module_base/module_container/ATen/ops/test/CMakeLists.txt b/source/module_base/module_container/ATen/ops/test/CMakeLists.txt
index 260aaa800f..0cc2c090ca 100644
--- a/source/module_base/module_container/ATen/ops/test/CMakeLists.txt
+++ b/source/module_base/module_container/ATen/ops/test/CMakeLists.txt
@@ -1,7 +1,7 @@
 AddTest(
   TARGET container_ops_uts
   LIBS parameter ${math_libs} 
-  SOURCES einsum_op_test.cpp linalg_op_test.cpp
+  SOURCES einsum_op_test.cpp linalg_op_test.cpp ../../kernels/lapack.cpp
 )
 
 target_link_libraries(container_ops_uts container base device)
\ No newline at end of file
diff --git a/source/module_elecstate/test/CMakeLists.txt b/source/module_elecstate/test/CMakeLists.txt
index 6a10494d4d..350e89153f 100644
--- a/source/module_elecstate/test/CMakeLists.txt
+++ b/source/module_elecstate/test/CMakeLists.txt
@@ -33,7 +33,7 @@ AddTest(
 AddTest(
   TARGET elecstate_fp_energy
   LIBS parameter  ${math_libs} base device
-  SOURCES elecstate_fp_energy_test.cpp ../fp_energy.cpp
+  SOURCES elecstate_fp_energy_test.cpp ../fp_energy.cpp ../elecstate_energy_terms.cpp
 )
 
 AddTest(
@@ -64,7 +64,7 @@ AddTest(
 AddTest(
   TARGET elecstate_energy
   LIBS parameter  ${math_libs} base device 
-  SOURCES elecstate_energy_test.cpp ../elecstate_energy.cpp ../fp_energy.cpp
+  SOURCES elecstate_energy_test.cpp ../elecstate_energy.cpp ../fp_energy.cpp ../elecstate_energy_terms.cpp
 )
 
 AddTest(
diff --git a/source/module_elecstate/test/elecstate_base_test.cpp b/source/module_elecstate/test/elecstate_base_test.cpp
index 02905c64ed..cbbd1aac68 100644
--- a/source/module_elecstate/test/elecstate_base_test.cpp
+++ b/source/module_elecstate/test/elecstate_base_test.cpp
@@ -1,5 +1,7 @@
 #include <string>
 
+#include"module_cell/setup_nonlocal.h"
+
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 #define private public
diff --git a/source/module_elecstate/test/elecstate_print_test.cpp b/source/module_elecstate/test/elecstate_print_test.cpp
index 0fb0248e64..dfb7e94559 100644
--- a/source/module_elecstate/test/elecstate_print_test.cpp
+++ b/source/module_elecstate/test/elecstate_print_test.cpp
@@ -1,5 +1,7 @@
 #include <string>
 
+#include"module_cell/setup_nonlocal.h"
+
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 #define private public