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

Commit 34e4f71

Browse files
ChaiBapchyamarcoabreu
authored andcommitted
Move imagenet inference to nightly (#16577)
* 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
1 parent ca4af0e commit 34e4f71

File tree

3 files changed

+18
-3
lines changed

3 files changed

+18
-3
lines changed

ci/docker/runtime_functions.sh

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1462,6 +1462,14 @@ nightly_test_installation() {
14621462
source ./tests/jenkins/run_test_installation_docs.sh docs/install/index.md 1 1686; ${1}
14631463
}
14641464

1465+
# Runs Imagenet inference
1466+
nightly_test_imagenet_inference() {
1467+
set -ex
1468+
echo $PWD
1469+
cp /work/mxnet/build/cpp-package/example/imagenet_inference .
1470+
/work/mxnet/cpp-package/example/inference/unit_test_imagenet_inference.sh
1471+
}
1472+
14651473
#Runs a simple MNIST training example
14661474
nightly_test_image_classification() {
14671475
set -ex

cpp-package/tests/ci_test.sh

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,6 @@ cp ../../build/cpp-package/example/test_regress_label .
6666
sh unittests/unit_test_mlp_csv.sh
6767

6868
cd inference
69-
cp ../../../build/cpp-package/example/imagenet_inference .
70-
./unit_test_imagenet_inference.sh
7169

7270
cp ../../../build/cpp-package/example/sentiment_analysis_rnn .
7371
./unit_test_sentiment_analysis_rnn.sh

tests/nightly/JenkinsfileForBinaries

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020

2121
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'
2222
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'
23+
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'
2324

2425
node('utility') {
2526
// Loading the utilities requires a node context unfortunately
@@ -36,7 +37,7 @@ core_logic: {
3637
ws('workspace/build-gpu') {
3738
utils.init_git()
3839
utils.docker_run('ubuntu_build_cuda', 'build_ubuntu_gpu_cuda101_cudnn7', false)
39-
utils.pack_lib('gpu', mx_lib)
40+
utils.pack_lib('gpu', mx_lib_cpp_example)
4041
}
4142
}
4243
}/*,
@@ -69,6 +70,14 @@ core_logic: {
6970
}
7071
}
7172
},
73+
'ImageNet Inference: GPU': {
74+
node(NODE_LINUX_GPU) {
75+
ws('workspace/nt-ImageInferenceTest') {
76+
utils.unpack_and_init('gpu', mx_lib_cpp_example)
77+
utils.docker_run('ubuntu_nightly_gpu', 'nightly_test_imagenet_inference', true)
78+
}
79+
}
80+
},
7281
'KVStore_SingleNode: GPU': {
7382
node('mxnetlinux-gpu-p3-8xlarge') {
7483
ws('workspace/nt-KVStoreTest') {

0 commit comments

Comments
 (0)