@@ -60,7 +60,7 @@ TEST(TestApiOps, div) {
60
60
for (auto i = 0 ; i < 64 ; i++) {
61
61
for (auto j = 0 ; j < 32 ; j++) {
62
62
for (auto k = 0 ; k < 32 ; k++) {
63
- ASSERT_EQ (arg1[i][j][k] / arg2[i][j][k], arg3[i][j][k]);
63
+ ASSERT_FLOAT_EQ (arg1[i][j][k] / arg2[i][j][k], arg3[i][j][k]);
64
64
}
65
65
}
66
66
}
@@ -113,7 +113,7 @@ TEST(TestApiOps, matMul) {
113
113
// Compare the results
114
114
for (auto i = 0 ; i < 512 ; i++) {
115
115
for (auto j = 0 ; j < 64 ; j++) {
116
- ASSERT_EQ (expected[i][j], argOut[i][j]);
116
+ ASSERT_FLOAT_EQ (expected[i][j], argOut[i][j]);
117
117
}
118
118
}
119
119
}
@@ -135,7 +135,7 @@ TEST(TestApiOps, mul) {
135
135
exec (" mul.json" , inputs, outputs);
136
136
137
137
for (auto i = 0 ; i < 128 ; i++) {
138
- ASSERT_EQ (arg3[i], static_cast <std::float32_t >(i * i));
138
+ ASSERT_FLOAT_EQ (arg3[i], static_cast <std::float32_t >(i * i));
139
139
}
140
140
}
141
141
@@ -157,7 +157,7 @@ TEST(TestApiOps, sub) {
157
157
exec (" sub.json" , inputs, outputs);
158
158
159
159
for (auto i = 0 ; i < 128 ; i++) {
160
- ASSERT_EQ (arg3[i], arg1[i] - arg2[i]);
160
+ ASSERT_FLOAT_EQ (arg3[i], arg1[i] - arg2[i]);
161
161
}
162
162
}
163
163
@@ -176,7 +176,7 @@ TEST(TestApiOps, pow) {
176
176
exec (" pow.json" , inputs, outputs);
177
177
178
178
for (auto i = 0 ; i < 64 ; i++) {
179
- ASSERT_EQ (arg1[i] * arg1[i], arg2[i]);
179
+ ASSERT_FLOAT_EQ (arg1[i] * arg1[i], arg2[i]);
180
180
}
181
181
}
182
182
@@ -196,7 +196,7 @@ TEST(TestApiOps, relu) {
196
196
exec (" relu.json" , inputs, outputs);
197
197
198
198
for (auto i = 0 ; i < 128 ; i++) {
199
- ASSERT_EQ (arg1[i] < 0 ? 0 : arg1[i], arg2[i]);
199
+ ASSERT_FLOAT_EQ (arg1[i] < 0 ? 0 : arg1[i], arg2[i]);
200
200
}
201
201
}
202
202
@@ -233,7 +233,7 @@ TEST(TestApiOps, reduceMean) {
233
233
234
234
for (auto x = 0 ; x < 16 ; x++) {
235
235
for (auto z = 0 ; z < 32 ; z++) {
236
- ASSERT_EQ (expected[x][0 ][z], arg2[x][0 ][z]);
236
+ ASSERT_FLOAT_EQ (expected[x][0 ][z], arg2[x][0 ][z]);
237
237
}
238
238
}
239
239
}
@@ -270,7 +270,7 @@ TEST(TestApiOps, reduceSum) {
270
270
271
271
for (auto x = 0 ; x < 16 ; x++) {
272
272
for (auto z = 0 ; z < 32 ; z++) {
273
- ASSERT_EQ (expected[x][0 ][z], arg2[x][0 ][z]);
273
+ ASSERT_FLOAT_EQ (expected[x][0 ][z], arg2[x][0 ][z]);
274
274
}
275
275
}
276
276
}
@@ -290,7 +290,7 @@ TEST(TestApiOps, sigmoid) {
290
290
exec (" sigmoid.json" , inputs, outputs);
291
291
292
292
for (auto i = 0 ; i < 128 ; i++) {
293
- ASSERT_EQ (1 .f / (1 .f + std::exp (-arg1[i])), arg2[i]);
293
+ ASSERT_FLOAT_EQ (1 .f / (1 .f + std::exp (-arg1[i])), arg2[i]);
294
294
}
295
295
}
296
296
0 commit comments