Skip to content
This repository was archived by the owner on Nov 25, 2023. It is now read-only.

Commit 966cfdd

Browse files
committed
add tnn model convertor build to ci
1 parent 6858db5 commit 966cfdd

File tree

2 files changed

+37
-1
lines changed

2 files changed

+37
-1
lines changed

.github/workflows/main.yml

+12-1
Original file line numberDiff line numberDiff line change
@@ -26,20 +26,22 @@ jobs:
2626
- name: Run a one-line script
2727
run: echo Hello, world!
2828

29+
2930
# Runs a set of commands using the runners shell
3031
- name: Run a multi-line script
3132
run: |
3233
echo Add other actions to build,
3334
echo test, and deploy your project.
3435
36+
3537
- name: pre-commit check
3638
run:
3739
cd .github/workflows && ./pre-commit-job.sh
3840

41+
3942
- name: python unit test
4043
run:
4144
.github/workflows/unit-test-job.sh
42-
4345

4446

4547
tnn:
@@ -49,6 +51,15 @@ jobs:
4951
- name: Run tnn build
5052
run: cd tnn && ./build.sh
5153

54+
55+
tnn-model-convertor:
56+
runs-on: ubuntu-latest
57+
steps:
58+
- uses: actions/checkout@v2
59+
- name: Run tnn model convertor build
60+
run: cd tnn && ./build_tnn_model_convertor.sh
61+
62+
5263
models:
5364
runs-on: ubuntu-latest
5465
steps:

tnn/build_tnn_model_convertor.sh

+25
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
#!/usr/bin/env bash
2+
set -ex
3+
4+
function prepare_env() {
5+
# default
6+
pip3 install onnx==1.6.0 onnxruntime numpy onnx-simplifier
7+
8+
# optional: tensorflow
9+
pip3 install tensorflow==1.15.0
10+
pip3 install tf2onnx
11+
pip3 install onnxruntime
12+
13+
# optional: caffe
14+
sudo apt-get install libprotobuf-dev protobuf-compiler
15+
}
16+
17+
function main() {
18+
prepare_env
19+
20+
# build
21+
cd tnn/tools/convert2tnn
22+
./build.sh
23+
}
24+
25+
main

0 commit comments

Comments
 (0)