diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index 8085c04d5eb7ce87f92da76b2ac04f204746cf6f..72820f2b14e09efd74226d35c23d72ae27f7cdec 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 615e63264c24a1e0c992acfb3f1e9f8d6076b715..37f7d960ec8e51b5eab93069a15562ef58d99dc0 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