From 44b4157688c31615cf3e5b0b9fb906aa93613558 Mon Sep 17 00:00:00 2001 From: Rohit Kumar Srivastava Date: Sun, 9 Feb 2020 21:08:36 -0800 Subject: [PATCH] enabling build stage gpu_int64 to enable large tensor nightly runs (#17546) --- tests/nightly/JenkinsfileForBinaries | 30 ++++++---------------------- 1 file changed, 6 insertions(+), 24 deletions(-) diff --git a/tests/nightly/JenkinsfileForBinaries b/tests/nightly/JenkinsfileForBinaries index c4ab5378a74c..1a7ef1aaaa05 100755 --- a/tests/nightly/JenkinsfileForBinaries +++ b/tests/nightly/JenkinsfileForBinaries @@ -40,25 +40,16 @@ core_logic: { utils.pack_lib('gpu', mx_lib_cpp_example_mkl) } } - }/*, + }, 'CPU: USE_INT64_TENSOR_SIZE': { - node(NODE_LINUX_CPU) { + node(NODE_LINUX_GPU) { ws('workspace/build-cpu-int64') { utils.init_git() utils.docker_run('ubuntu_nightly_cpu', 'build_ubuntu_cpu_large_tensor', false) utils.pack_lib('cpu_int64', mx_cmake_lib) } } - }, - 'GPU: USE_INT64_TENSOR_SIZE': { - node(NODE_LINUX_GPU) { - ws('workspace/build-gpu-int64') { - utils.init_git() - utils.docker_run('ubuntu_nightly_gpu', 'build_ubuntu_gpu_large_tensor', true) - utils.pack_lib('gpu_int64', mx_cmake_lib) - } - } - }*/ + } } stage('NightlyTests'){ @@ -86,21 +77,12 @@ core_logic: { } } }, - // https://github.com/apache/incubator-mxnet/issues/14980 - /*'Test Large Tensor Size: CPU': { - node(NODE_LINUX_CPU) { + 'Test Large Tensor Size: CPU': { + node(NODE_LINUX_GPU) { ws('workspace/large_tensor-cpu') { utils.unpack_and_init('cpu_int64', mx_cmake_lib) utils.docker_run('ubuntu_nightly_cpu', 'nightly_test_large_tensor', false) - } - } - },*/ - 'Test Large Tensor Size: GPU': { - node(NODE_LINUX_GPU) { - ws('workspace/large_tensor-gpu') { - utils.unpack_and_init('gpu_int64', mx_cmake_lib) - utils.docker_run('ubuntu_nightly_gpu', 'nightly_test_large_tensor', true) - utils.docker_run('ubuntu_nightly_gpu', 'nightly_test_large_vector', true) + utils.docker_run('ubuntu_nightly_cpu', 'nightly_test_large_vector', false) } } },