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

Commit

Permalink
Move imagenet inference to nightly (#16577)
Browse files Browse the repository at this point in the history
* move to nightly

* bracket fix

* test imgnet comment rest

* br

* change dir

* change lib

* uncomment other tests

* change lib for build

* remove extraneous libs

* add cd to access shell script

* ready to merge

* fix dir structure
  • Loading branch information
ChaiBapchya authored and marcoabreu committed Oct 22, 2019
1 parent ca4af0e commit 34e4f71
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 3 deletions.
8 changes: 8 additions & 0 deletions ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -1462,6 +1462,14 @@ nightly_test_installation() {
source ./tests/jenkins/run_test_installation_docs.sh docs/install/index.md 1 1686; ${1}
}

# Runs Imagenet inference
nightly_test_imagenet_inference() {
set -ex
echo $PWD
cp /work/mxnet/build/cpp-package/example/imagenet_inference .
/work/mxnet/cpp-package/example/inference/unit_test_imagenet_inference.sh
}

#Runs a simple MNIST training example
nightly_test_image_classification() {
set -ex
Expand Down
2 changes: 0 additions & 2 deletions cpp-package/tests/ci_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -66,8 +66,6 @@ cp ../../build/cpp-package/example/test_regress_label .
sh unittests/unit_test_mlp_csv.sh

cd inference
cp ../../../build/cpp-package/example/imagenet_inference .
./unit_test_imagenet_inference.sh

cp ../../../build/cpp-package/example/sentiment_analysis_rnn .
./unit_test_sentiment_analysis_rnn.sh
Expand Down
11 changes: 10 additions & 1 deletion tests/nightly/JenkinsfileForBinaries
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@

mx_lib = 'lib/libmxnet.so, lib/libmxnet.a, lib/libtvm_runtime.so, lib/libtvmop.so, 3rdparty/dmlc-core/libdmlc.a, 3rdparty/tvm/nnvm/lib/libnnvm.a'
mx_cmake_lib = 'build/libmxnet.so, build/libmxnet.a, build/3rdparty/tvm/libtvm_runtime.so, build/libtvmop.so, build/3rdparty/dmlc-core/libdmlc.a, build/tests/mxnet_unit_tests, build/3rdparty/openmp/runtime/src/libomp.so'
mx_lib_cpp_example = 'lib/libmxnet.so, lib/libmxnet.a, lib/libtvm_runtime.so, lib/libtvmop.so, 3rdparty/dmlc-core/libdmlc.a, 3rdparty/tvm/nnvm/lib/libnnvm.a, build/cpp-package/example/imagenet_inference'

node('utility') {
// Loading the utilities requires a node context unfortunately
Expand All @@ -36,7 +37,7 @@ core_logic: {
ws('workspace/build-gpu') {
utils.init_git()
utils.docker_run('ubuntu_build_cuda', 'build_ubuntu_gpu_cuda101_cudnn7', false)
utils.pack_lib('gpu', mx_lib)
utils.pack_lib('gpu', mx_lib_cpp_example)
}
}
}/*,
Expand Down Expand Up @@ -69,6 +70,14 @@ core_logic: {
}
}
},
'ImageNet Inference: GPU': {
node(NODE_LINUX_GPU) {
ws('workspace/nt-ImageInferenceTest') {
utils.unpack_and_init('gpu', mx_lib_cpp_example)
utils.docker_run('ubuntu_nightly_gpu', 'nightly_test_imagenet_inference', true)
}
}
},
'KVStore_SingleNode: GPU': {
node('mxnetlinux-gpu-p3-8xlarge') {
ws('workspace/nt-KVStoreTest') {
Expand Down

0 comments on commit 34e4f71

Please sign in to comment.