OPTFIX2

Critsium-xy · Critsium-xy · commit f558e070df50 · 2025-01-03T13:51:36.000+08:00
diff --git a/source/module_base/opt_CG.h b/source/module_base/opt_CG.h
@@ -5,7 +5,11 @@
 
 #include <iostream>
 
+#ifdef __CUDA
+#undef __CUDA
 #include "global_function.h"
+#define __CUDA
+#endif
 #include "module_base/parallel_reduce.h"
 
 namespace ModuleBase
diff --git a/source/module_base/test/CMakeLists.txt b/source/module_base/test/CMakeLists.txt
@@ -144,13 +144,13 @@ AddTest(
 
 AddTest(
   TARGET base_opt_CG
-  LIBS parameter  ${math_libs} device
+  LIBS parameter  ${math_libs}
   SOURCES opt_CG_test.cpp opt_test_tools.cpp ../blas_connector.cpp ../opt_CG.cpp ../opt_DCsrch.cpp ../global_variable.cpp ../parallel_reduce.cpp
 )
 
 AddTest(
   TARGET base_opt_TN
-  LIBS parameter  ${math_libs} device
+  LIBS parameter  ${math_libs}
   SOURCES opt_TN_test.cpp opt_test_tools.cpp ../blas_connector.cpp ../opt_CG.cpp ../opt_DCsrch.cpp ../global_variable.cpp ../parallel_reduce.cpp
 )
 

Original file line number	Diff line number	Diff line change
`@@ -144,13 +144,13 @@ AddTest(`
`144`	`144`
`145`	`145`	`AddTest(`
`146`	`146`	`TARGET base_opt_CG`
`147`		`- LIBS parameter ${math_libs} device`
	`147`	`+ LIBS parameter ${math_libs}`
`148`	`148`	`SOURCES opt_CG_test.cpp opt_test_tools.cpp ../blas_connector.cpp ../opt_CG.cpp ../opt_DCsrch.cpp ../global_variable.cpp ../parallel_reduce.cpp`
`149`	`149`	`)`
`150`	`150`
`151`	`151`	`AddTest(`
`152`	`152`	`TARGET base_opt_TN`
`153`		`- LIBS parameter ${math_libs} device`
	`153`	`+ LIBS parameter ${math_libs}`
`154`	`154`	`SOURCES opt_TN_test.cpp opt_test_tools.cpp ../blas_connector.cpp ../opt_CG.cpp ../opt_DCsrch.cpp ../global_variable.cpp ../parallel_reduce.cpp`
`155`	`155`	`)`
`156`	`156`