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

Commit

Permalink
Add cython build to CI
Browse files Browse the repository at this point in the history
  • Loading branch information
asitstands committed Jun 16, 2018
1 parent df1a291 commit 0000950
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 4 deletions.
10 changes: 6 additions & 4 deletions Jenkinsfile
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,8 @@

// mxnet libraries
mx_lib = 'lib/libmxnet.so, lib/libmxnet.a, 3rdparty/dmlc-core/libdmlc.a, 3rdparty/tvm/nnvm/lib/libnnvm.a'
// mxnet libraries with cython
mx_cython_lib = 'lib/libmxnet.so, lib/libmxnet.a, python/mxnet/_cy2/*.so, python/mxnet/_cy3/*.so, 3rdparty/dmlc-core/libdmlc.a, 3rdparty/tvm/nnvm/lib/libnnvm.a'
// for scala build, need to pass extra libs when run with dist_kvstore
mx_dist_lib = 'lib/libmxnet.so, lib/libmxnet.a, 3rdparty/dmlc-core/libdmlc.a, 3rdparty/tvm/nnvm/lib/libnnvm.a, 3rdparty/ps-lite/build/libps.a, deps/lib/libprotobuf-lite.a, deps/lib/libzmq.a'
// mxnet cmake libraries, in cmake builds we do not produce a libnvvm static library by default.
Expand Down Expand Up @@ -170,7 +172,7 @@ try {
timeout(time: max_time, unit: 'MINUTES') {
init_git()
docker_run('centos7_cpu', 'build_centos7_cpu', false)
pack_lib('centos7_cpu')
pack_lib('centos7_cpu', mx_cython_lib)
}
}
}
Expand All @@ -192,7 +194,7 @@ try {
timeout(time: max_time, unit: 'MINUTES') {
init_git()
docker_run('centos7_gpu', 'build_centos7_gpu', false)
pack_lib('centos7_gpu')
pack_lib('centos7_gpu', mx_cython_lib)
}
}
}
Expand Down Expand Up @@ -632,7 +634,7 @@ try {
timeout(time: max_time, unit: 'MINUTES') {
try {
init_git()
unpack_lib('centos7_cpu')
unpack_lib('centos7_cpu', mx_cython_lib)
docker_run('centos7_cpu', 'unittest_centos7_cpu', false)
} finally {
collect_test_results_unix('nosetests_unittest.xml', 'nosetests_python3_centos7_cpu_unittest.xml')
Expand All @@ -648,7 +650,7 @@ try {
timeout(time: max_time, unit: 'MINUTES') {
try {
init_git()
unpack_lib('centos7_gpu')
unpack_lib('centos7_gpu', mx_cython_lib)
docker_run('centos7_gpu', 'unittest_centos7_gpu', true)
} finally {
collect_test_results_unix('nosetests_gpu.xml', 'nosetests_python3_centos7_gpu.xml')
Expand Down
4 changes: 4 additions & 0 deletions ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -267,6 +267,7 @@ build_centos7_cpu() {
USE_BLAS=openblas \
USE_DIST_KVSTORE=1 \
-j$(nproc)
make cython
}

build_centos7_mkldnn() {
Expand Down Expand Up @@ -298,6 +299,7 @@ build_centos7_gpu() {
USE_CUDNN=1 \
USE_DIST_KVSTORE=1 \
-j$(nproc)
make cython
}

build_ubuntu_cpu() {
Expand Down Expand Up @@ -633,13 +635,15 @@ unittest_ubuntu_gpu_R() {
unittest_centos7_cpu() {
set -ex
cd /work/mxnet
export MXNET_ENFORCE_CYTHON=1
python3.6 -m "nose" --with-xunit --xunit-file nosetests_unittest.xml --verbose tests/python/unittest
python3.6 -m "nose" --with-xunit --xunit-file nosetests_train.xml --verbose tests/python/train
}

unittest_centos7_gpu() {
set -ex
cd /work/mxnet
export MXNET_ENFORCE_CYTHON=1
python3.6 -m "nose" --with-xunit --xunit-file nosetests_gpu.xml --verbose tests/python/gpu
}

Expand Down

0 comments on commit 0000950

Please sign in to comment.