提交 8749f4fc 编写于 作者: C Chunwei

Enhance gitlab ci

...@@ -9,6 +9,8 @@ stages: ...@@ -9,6 +9,8 @@ stages:
- build_mobile - build_mobile
check:prebuilt: check:prebuilt:
tags:
- lite
stage: ci stage: ci
script: script:
#- pip3 install pre-commit #- pip3 install pre-commit
...@@ -24,17 +26,21 @@ check:prebuilt: ...@@ -24,17 +26,21 @@ check:prebuilt:
- /root/.cache - /root/.cache
build:server: build:server:
tags:
- lite
image: $SERVER_LITE_DOCKER_IMAGE image: $SERVER_LITE_DOCKER_IMAGE
stage: build_server stage: build_server
cache: cache:
key: server_thirdparty key: server_thirdparty
paths: paths:
- build/third_party - build/third_party
- /root/.ccache
script: script:
#- export http_proxy=http://172.19.57.45:3128 - apt install ccache
#- export https_proxy=http://172.19.57.45:3128 - export http_proxy=http://172.19.57.45:3128
- export http_proxy=http://agent.baidu.com:8118 - export https_proxy=http://172.19.57.45:3128
- export https_proxy=http://agent.baidu.com:8118 #- export http_proxy=http://agent.baidu.com:8118
#- export https_proxy=http://agent.baidu.com:8118
- mkdir -p build - mkdir -p build
- cd build - cd build
- ../paddle/fluid/lite/tools/build.sh cmake_x86 - ../paddle/fluid/lite/tools/build.sh cmake_x86
...@@ -49,6 +55,8 @@ build:server: ...@@ -49,6 +55,8 @@ build:server:
- check:prebuilt - check:prebuilt
build:mobile: build:mobile:
tags:
- lite
stage: build_mobile stage: build_mobile
image: $MOBILE_LITE_DOCKER_IMAGE image: $MOBILE_LITE_DOCKER_IMAGE
cache: cache:
...@@ -56,7 +64,9 @@ build:mobile: ...@@ -56,7 +64,9 @@ build:mobile:
paths: paths:
- $MOBILE_LITE_CACHE0 - $MOBILE_LITE_CACHE0
- $MOBILE_LITE_CACHE1 - $MOBILE_LITE_CACHE1
- /root/.ccache
script: script:
- apt install ccache
- export http_proxy=http://172.19.57.45:3128 - export http_proxy=http://172.19.57.45:3128
- export https_proxy=http://172.19.57.45:3128 - export https_proxy=http://172.19.57.45:3128
- ./paddle/fluid/lite/tools/build.sh build_test_arm - ./paddle/fluid/lite/tools/build.sh build_test_arm
......
...@@ -166,6 +166,7 @@ if (WITH_LITE AND LITE_WITH_LIGHT_WEIGHT_FRAMEWORK) ...@@ -166,6 +166,7 @@ if (WITH_LITE AND LITE_WITH_LIGHT_WEIGHT_FRAMEWORK)
#include(external/zlib) # download, build, install gtest #include(external/zlib) # download, build, install gtest
include(external/protobuf) # download, build, install protobuf include(external/protobuf) # download, build, install protobuf
include(external/eigen) # download eigen3 include(external/eigen) # download eigen3
include(ccache) # set ccache for compilation
include(generic) # simplify cmake module include(generic) # simplify cmake module
include(configure) # add paddle env configuration include(configure) # add paddle env configuration
......
...@@ -18,10 +18,10 @@ import numpy as np ...@@ -18,10 +18,10 @@ import numpy as np
import paddle.fluid as fluid import paddle.fluid as fluid
from paddle.fluid.backward import append_backward from paddle.fluid.backward import append_backward
a = fluid.layers.data(name="a", shape=[100], dtype='float32') a = fluid.layers.data(name="a", shape=[2], dtype='float32')
label = fluid.layers.data(name="label", shape=[100], dtype='float32') label = fluid.layers.data(name="label", shape=[10], dtype='float32')
a1 = fluid.layers.fc(input=a, size=500, act=None, bias_attr=False) a1 = fluid.layers.fc(input=a, size=3, act=None, bias_attr=False)
cost = fluid.layers.square_error_cost(a1, label) cost = fluid.layers.square_error_cost(a1, label)
avg_cost = fluid.layers.mean(cost) avg_cost = fluid.layers.mean(cost)
...@@ -36,7 +36,7 @@ exe.run(fluid.default_startup_program()) ...@@ -36,7 +36,7 @@ exe.run(fluid.default_startup_program())
with open('startup_program.pb', 'wb') as f: with open('startup_program.pb', 'wb') as f:
f.write(fluid.default_startup_program().desc.serialize_to_string()) f.write(fluid.default_startup_program().desc.serialize_to_string())
data_1 = np.array(numpy.random.random([100, 100]), dtype='float32') #data_1 = np.array(numpy.random.random([100, 100]), dtype='float32')
#fluid.default_main_program().desc. #fluid.default_main_program().desc.
...@@ -50,7 +50,7 @@ with open('main_program.pb', 'wb') as f: ...@@ -50,7 +50,7 @@ with open('main_program.pb', 'wb') as f:
#outs = exe.run(program=prog, feed={'a':data_1, }, fetch_list=[cost]) #outs = exe.run(program=prog, feed={'a':data_1, }, fetch_list=[cost])
sys.exit(0) #sys.exit(0)
fluid.io.save_inference_model("./model2", [a.name], [a1], exe) fluid.io.save_inference_model("./model2", [a.name], [a1], exe)
print(numpy.array(outs)) #print(numpy.array(outs))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册