From 9a310e59e9395d116d2dc5043574aa5f17a0bc2f Mon Sep 17 00:00:00 2001
From: Marco de Abreu <marcoabreu@users.noreply.github.com>
Date: Wed, 21 Mar 2018 18:03:14 +0100
Subject: [PATCH] Reformat Jenkinsfile and switch quantization to CUDA 9 (#9)

---
 Jenkinsfile | 33 ++++-----------------------------
 1 file changed, 4 insertions(+), 29 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index 5c99ab3be3a7..4778137803a3 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -121,20 +121,6 @@ def python3_gpu_ut(docker_container_name) {
   }
 }
 
-// Python 2 gpu quantization unittest
-def python2_gpu_qt(docker_container_name) {
-  timeout(time: max_time, unit: 'MINUTES') {
-    sh "ci/build.py --nvidiadocker --build --platform ${docker_container_name} /work/runtime_functions.sh unittest_ubuntu_python2_quantization_gpu"
-  }
-}
-
-// Python 3 gpu quantization unittest
-def python3_gpu_qt(docker_container_name) {
-  timeout(time: max_time, unit: 'MINUTES') {
-    sh "ci/build.py --nvidiadocker --build --platform ${docker_container_name} /work/runtime_functions.sh unittest_ubuntu_python3_quantization_gpu"
-  }
-}
-
 try {
   stage("Sanity Check") {
     node('mxnetlinux-cpu') {
@@ -218,17 +204,6 @@ try {
         }
       }
     },
-    'GPU: CUDA8.0+cuDNN7': {
-      node('mxnetlinux-cpu') {
-        ws('workspace/build-gpu-cuda8-cudnn7') {
-          init_git()
-          sh "ci/build.py --build --platform ubuntu_build_cuda /work/runtime_functions.sh build_ubuntu_gpu_cuda8_cudnn7"
-          pack_lib('gpu_cuda8_cudnn7')
-          stash includes: 'build/cpp-package/example/test_score', name: 'cpp_test_score'
-          stash includes: 'build/cpp-package/example/test_optimizer', name: 'cpp_test_optimizer'
-        }
-      }
-    },
     'Amalgamation MIN': {
       node('mxnetlinux-cpu') {
         ws('workspace/amalgamationmin') {
@@ -389,8 +364,8 @@ try {
       node('mxnetlinux-gpu-p3') {
         ws('workspace/ut-python2-quantize-gpu') {
           init_git()
-          unpack_lib('gpu_cuda8_cudnn7', mx_lib)
-          python2_gpu_qt('ubuntu_gpu')
+          unpack_lib('gpu', mx_lib)
+          sh "ci/build.py --nvidiadocker --build --platform ubuntu_gpu /work/runtime_functions.sh unittest_ubuntu_python2_quantization_gpu"
         }
       }
     },
@@ -398,8 +373,8 @@ try {
       node('mxnetlinux-gpu-p3') {
         ws('workspace/ut-python3-quantize-gpu') {
           init_git()
-          unpack_lib('gpu_cuda8_cudnn7', mx_lib)
-          python3_gpu_qt('ubuntu_gpu')
+          unpack_lib('gpu', mx_lib)
+          sh "ci/build.py --nvidiadocker --build --platform ubuntu_gpu /work/runtime_functions.sh unittest_ubuntu_python3_quantization_gpu"
         }
       }
     },