From 12388399ee1e37c2172744c9f6409e968f545cb4 Mon Sep 17 00:00:00 2001 From: Rohit Kumar Srivastava Date: Fri, 7 Feb 2020 20:03:21 +0000 Subject: [PATCH] enabling build stage gpu_int64 to enable large tensor nightly runs --- tests/nightly/JenkinsfileForBinaries | 29 ++++++---------------------- 1 file changed, 6 insertions(+), 23 deletions(-) diff --git a/tests/nightly/JenkinsfileForBinaries b/tests/nightly/JenkinsfileForBinaries index c4ab5378a74c..3fc7486debfb 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'){ @@ -87,20 +78,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) } } },