From 65e77beffcdba372d5ccda94da8640df7e49ebc4 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Tue, 26 Jun 2018 14:35:09 +0800
Subject: [PATCH] Add python tools test to CI.

---
 .gitlab-ci.yml                             | 13 ++++++++++++-
 mace/ops/winograd_convolution_benchmark.cc |  2 --
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 8085c04d..72820f2b 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -6,6 +6,7 @@ stages:
   - ndk_versions_compatible_tests
   - ops_test
   - api_test
+  - python_tools_tests
   - ops_benchmark
   - extra_tests
 
@@ -91,4 +92,14 @@ ndk_versions_compatible_tests:
       fi
       done
     - export ANDROID_NDK_HOME=$DEFAULT_NDK_PATH
-    - export PATH=$ANDROID_NDK_HOME:$PATH
\ No newline at end of file
+    - export PATH=$ANDROID_NDK_HOME:$PATH
+
+python_tools_tests:
+  stage: python_tools_tests
+  script:
+    - pwd
+    - rm -rf mace-models
+    - GIT_SSH_COMMAND="ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no" git clone git@v9.git.n.xiaomi.com:deep-computing/mace-models.git
+    - CONF_FILE=mace-models/mobilenet-v2/mobilenet-v2.yml
+    - sh -c "python tools/converter.py build --config=${CONF_FILE} --disable_tuning && python tools/converter.py run --config=${CONF_FILE} --round=1 --validate" || exit 1
+    - rm -rf mace-models
diff --git a/mace/ops/winograd_convolution_benchmark.cc b/mace/ops/winograd_convolution_benchmark.cc
index 615e6326..37f7d960 100644
--- a/mace/ops/winograd_convolution_benchmark.cc
+++ b/mace/ops/winograd_convolution_benchmark.cc
@@ -115,8 +115,6 @@ MACE_BM_WINOGRAD_CONV(1, 56, 56, 256, 256, 2);
 MACE_BM_WINOGRAD_CONV(1, 56, 56, 256, 256, 4);
 MACE_BM_WINOGRAD_CONV(1, 128, 128, 128, 256, 2);
 MACE_BM_WINOGRAD_CONV(1, 128, 128, 128, 256, 4);
-MACE_BM_WINOGRAD_CONV(1, 256, 256, 256, 256, 2);
-MACE_BM_WINOGRAD_CONV(1, 256, 256, 256, 256, 4);
 
 }  // namespace test
 }  // namespace ops
-- 
GitLab