Skip to content

Commit 0836ebe

Browse files
authored
Merge pull request #13677 from JiabinYang/mac/fix_unittest
Cherry-pick from develop to fix unittest on Mac
2 parents f6a8b67 + 8c65a2f commit 0836ebe

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

paddle/fluid/inference/api/api_impl_tester.cc

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,12 @@ limitations under the License. */
2121
#include "paddle/fluid/inference/api/api_impl.h"
2222
#include "paddle/fluid/inference/tests/test_helper.h"
2323

24+
#ifdef __clang__
25+
#define ACC_DIFF 4e-3
26+
#else
27+
#define ACC_DIFF 1e-3
28+
#endif
29+
2430
DEFINE_string(dirname, "", "Directory of the inference model.");
2531

2632
namespace paddle {
@@ -99,8 +105,8 @@ void MainWord2Vec(bool use_gpu) {
99105

100106
float* lod_data = output1.data<float>();
101107
for (int i = 0; i < output1.numel(); ++i) {
102-
EXPECT_LT(lod_data[i] - data[i], 1e-3);
103-
EXPECT_GT(lod_data[i] - data[i], -1e-3);
108+
EXPECT_LT(lod_data[i] - data[i], ACC_DIFF);
109+
EXPECT_GT(lod_data[i] - data[i], -ACC_DIFF);
104110
}
105111
}
106112

@@ -144,7 +150,7 @@ void MainImageClassification(bool use_gpu) {
144150
float* data = static_cast<float*>(outputs[0].data.data());
145151
float* lod_data = output1.data<float>();
146152
for (size_t j = 0; j < len / sizeof(float); ++j) {
147-
EXPECT_NEAR(lod_data[j], data[j], 1e-3);
153+
EXPECT_NEAR(lod_data[j], data[j], ACC_DIFF);
148154
}
149155
}
150156

@@ -199,7 +205,7 @@ void MainThreadsWord2Vec(bool use_gpu) {
199205
float* ref_data = refs[tid].data<float>();
200206
EXPECT_EQ(refs[tid].numel(), static_cast<int64_t>(len / sizeof(float)));
201207
for (int i = 0; i < refs[tid].numel(); ++i) {
202-
EXPECT_NEAR(ref_data[i], data[i], 1e-3);
208+
EXPECT_NEAR(ref_data[i], data[i], ACC_DIFF);
203209
}
204210
});
205211
}
@@ -251,7 +257,7 @@ void MainThreadsImageClassification(bool use_gpu) {
251257
float* ref_data = refs[tid].data<float>();
252258
EXPECT_EQ((size_t)refs[tid].numel(), len / sizeof(float));
253259
for (int i = 0; i < refs[tid].numel(); ++i) {
254-
EXPECT_NEAR(ref_data[i], data[i], 1e-3);
260+
EXPECT_NEAR(ref_data[i], data[i], ACC_DIFF);
255261
}
256262
});
257263
}

0 commit comments

Comments
 (0)