From b226777ad31422430464a3a96ea91265f3f96622 Mon Sep 17 00:00:00 2001 From: "xiang.zhang" Date: Mon, 30 Aug 2021 16:15:21 +0800 Subject: [PATCH] Fix average_pool unit test failure: precesion issue Signed-off-by: xiang.zhang --- src/tim/vx/ops/avg_pool_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tim/vx/ops/avg_pool_test.cc b/src/tim/vx/ops/avg_pool_test.cc index 8838f3d..0465302 100644 --- a/src/tim/vx/ops/avg_pool_test.cc +++ b/src/tim/vx/ops/avg_pool_test.cc @@ -298,7 +298,7 @@ TEST(AVG_ANDROID, shape_60_52_3_5_fp32_kernel_35_stride_5) { std::vector output(golden.size()); EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data())); - EXPECT_EQ(golden, output); + ArraysMatch(golden, output, 1e-5f); } TEST(AVG_ANDROID, shape_60_52_3_5_uint8_kernel_35_stride_5) {