Skip to content

Commit 1cd1ffc

Browse files
committed
Fix MPI Bug
1 parent 8a8ec04 commit 1cd1ffc

File tree

3 files changed

+24
-0
lines changed

3 files changed

+24
-0
lines changed

source/module_base/test/math_chebyshev_test.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,6 +336,9 @@ TEST_F(MathChebyshevTest, tracepolyA)
336336

337337
TEST_F(MathChebyshevTest, checkconverge)
338338
{
339+
#ifdef __MPI
340+
#undef __MPI
341+
#endif
339342
const int norder = 100;
340343
p_chetest = new ModuleBase::Chebyshev<double>(norder);
341344
auto fun_sigma_y

source/module_base/test/opt_CG_test.cpp

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,9 @@ class CG_test : public testing::Test
143143

144144
TEST_F(CG_test, Stand_Solve_LinearEq)
145145
{
146+
#ifdef __MPI
147+
#undef __MPI
148+
#endif
146149
CG_Solve_LinearEq();
147150
EXPECT_NEAR(x[0], 0.5, DOUBLETHRESHOLD);
148151
EXPECT_NEAR(x[1], 1.6429086563584579739e-18, DOUBLETHRESHOLD);
@@ -153,6 +156,9 @@ TEST_F(CG_test, Stand_Solve_LinearEq)
153156

154157
TEST_F(CG_test, PR_Solve_LinearEq)
155158
{
159+
#ifdef __MPI
160+
#undef __MPI
161+
#endif
156162
Solve(1, 0);
157163
EXPECT_NEAR(x[0], 0.50000000000003430589, DOUBLETHRESHOLD);
158164
EXPECT_NEAR(x[1], -3.4028335704761047964e-14, DOUBLETHRESHOLD);
@@ -163,6 +169,9 @@ TEST_F(CG_test, PR_Solve_LinearEq)
163169

164170
TEST_F(CG_test, HZ_Solve_LinearEq)
165171
{
172+
#ifdef __MPI
173+
#undef __MPI
174+
#endif
166175
Solve(2, 0);
167176
EXPECT_NEAR(x[0], 0.49999999999999944489, DOUBLETHRESHOLD);
168177
EXPECT_NEAR(x[1], -9.4368957093138305936e-16, DOUBLETHRESHOLD);
@@ -173,6 +182,9 @@ TEST_F(CG_test, HZ_Solve_LinearEq)
173182

174183
TEST_F(CG_test, PR_Min_Func)
175184
{
185+
#ifdef __MPI
186+
#undef __MPI
187+
#endif
176188
Solve(1, 1);
177189
EXPECT_NEAR(x[0], 4.0006805979150792396, DOUBLETHRESHOLD);
178190
EXPECT_NEAR(x[1], 2.0713759992720870429, DOUBLETHRESHOLD);
@@ -183,6 +195,9 @@ TEST_F(CG_test, PR_Min_Func)
183195

184196
TEST_F(CG_test, HZ_Min_Func)
185197
{
198+
#ifdef __MPI
199+
#undef __MPI
200+
#endif
186201
Solve(2, 1);
187202
EXPECT_NEAR(x[0], 4.0006825378033568086, DOUBLETHRESHOLD);
188203
EXPECT_NEAR(x[1], 2.0691732100663737803, DOUBLETHRESHOLD);

source/module_base/test/opt_TN_test.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,9 @@ class TN_test : public testing::Test
110110

111111
TEST_F(TN_test, TN_Solve_LinearEq)
112112
{
113+
#ifdef __MPI
114+
#undef __MPI
115+
#endif
113116
Solve(0);
114117
EXPECT_NEAR(x[0], 0.50000000000003430589, DOUBLETHRESHOLD);
115118
EXPECT_NEAR(x[1], -3.4028335704761047964e-14, DOUBLETHRESHOLD);
@@ -120,6 +123,9 @@ TEST_F(TN_test, TN_Solve_LinearEq)
120123

121124
TEST_F(TN_test, TN_Min_Func)
122125
{
126+
#ifdef __MPI
127+
#undef __MPI
128+
#endif
123129
Solve(1);
124130
EXPECT_NEAR(x[0], 4.0049968540891525137, DOUBLETHRESHOLD);
125131
EXPECT_NEAR(x[1], 2.1208751163987624722, DOUBLETHRESHOLD);

0 commit comments

Comments
 (0)