提交 5b244d8d 编写于 作者: 李寅

Merge branch 'add-tools-test-to-ci' into 'master'

Add python tools test to CI.

See merge request !605
...@@ -6,6 +6,7 @@ stages: ...@@ -6,6 +6,7 @@ stages:
- ndk_versions_compatible_tests - ndk_versions_compatible_tests
- ops_test - ops_test
- api_test - api_test
- python_tools_tests
- ops_benchmark - ops_benchmark
- extra_tests - extra_tests
...@@ -91,4 +92,14 @@ ndk_versions_compatible_tests: ...@@ -91,4 +92,14 @@ ndk_versions_compatible_tests:
fi fi
done done
- export ANDROID_NDK_HOME=$DEFAULT_NDK_PATH - export ANDROID_NDK_HOME=$DEFAULT_NDK_PATH
- export PATH=$ANDROID_NDK_HOME:$PATH - export PATH=$ANDROID_NDK_HOME:$PATH
\ No newline at end of file
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
...@@ -115,8 +115,6 @@ MACE_BM_WINOGRAD_CONV(1, 56, 56, 256, 256, 2); ...@@ -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, 56, 56, 256, 256, 4);
MACE_BM_WINOGRAD_CONV(1, 128, 128, 128, 256, 2); 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, 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 test
} // namespace ops } // namespace ops
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册