Re-Enabling Large Tensor Nightly on GPU

apache · Feb 5, 2020 · 4e871db · 4e871db
1 parent 4054355
commit 4e871db
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/tests/nightly/JenkinsfileForBinaries b/tests/nightly/JenkinsfileForBinaries
@@ -96,14 +96,22 @@ core_logic: {
       }
     },*/
     // https://github.com/apache/incubator-mxnet/issues/14981
-    /*'Test Large Tensor Size: GPU': {
+    '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)
         }
       }
-    },*/
+    },
+    'Test Large Vector: 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_vector', true)
+        }
+      }
+    },
     'Gluon estimator: GPU': {
       node(NODE_LINUX_GPU) {
         ws('workspace/estimator-test-gpu') {