From 2386c0345b7511272b3f5c970c5e741a9c5d4a30 Mon Sep 17 00:00:00 2001 From: Yuan Shuai Date: Tue, 24 Mar 2020 20:09:12 +0800 Subject: [PATCH] [LITE][OPENCL] Fix opencl ci for mobilenetv1 v2 test (#3257) * fix ci for test_mobilenetv1/v2. test=develop * [OPENCL][CI] add test_mobilenetv1/v2 for opencl ci. test=develop --- lite/api/mobilenetv1_test.cc | 2 +- lite/api/mobilenetv2_test.cc | 2 +- lite/tools/ci_build.sh | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/lite/api/mobilenetv1_test.cc b/lite/api/mobilenetv1_test.cc index fff27c8156..5342a36ec1 100644 --- a/lite/api/mobilenetv1_test.cc +++ b/lite/api/mobilenetv1_test.cc @@ -95,7 +95,7 @@ void TestModel(const std::vector& valid_places, if (first_target == TARGET(kOpenCL) || first_target == TARGET(kNPU)) { ASSERT_EQ(out->dims().production(), 1000); - double eps = 0.1; + double eps = first_target == TARGET(kOpenCL) ? 0.12 : 0.1; for (int i = 0; i < ref.size(); ++i) { for (int j = 0; j < ref[i].size(); ++j) { auto result = pdata[j * step + (out->dims()[1] * i)]; diff --git a/lite/api/mobilenetv2_test.cc b/lite/api/mobilenetv2_test.cc index c2959bf88d..465f82056c 100644 --- a/lite/api/mobilenetv2_test.cc +++ b/lite/api/mobilenetv2_test.cc @@ -97,7 +97,7 @@ void TestModel(const std::vector& valid_places, if (first_target == TARGET(kOpenCL) || first_target == TARGET(kNPU)) { ASSERT_EQ(out->dims().production(), 1000); - double eps = 0.1; + double eps = first_target == TARGET(kOpenCL) ? 0.15 : 0.1; for (int i = 0; i < ref.size(); ++i) { for (int j = 0; j < ref[i].size(); ++j) { auto result = pdata[j * step + (out->dims()[1] * i)]; diff --git a/lite/tools/ci_build.sh b/lite/tools/ci_build.sh index 84ae37d4de..703da69fa5 100755 --- a/lite/tools/ci_build.sh +++ b/lite/tools/ci_build.sh @@ -1140,8 +1140,8 @@ function main { ;; build_test_arm_opencl) build_test_arm_opencl -# build_test_arm_subtask_model test_mobilenetv1 mobilenet_v1 -# build_test_arm_subtask_model test_mobilenetv2 mobilenet_v2_relu + build_test_arm_subtask_model test_mobilenetv1 mobilenet_v1 + build_test_arm_subtask_model test_mobilenetv2 mobilenet_v2_relu shift ;; build_test_arm_subtask_android) -- GitLab