From d4ca627a5a5df88f477bd6cc89ee2e3e06931c29 Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Tue, 11 Jun 2019 10:55:37 -0700
Subject: [PATCH] [CI] separate out legacy as a stage (#3337)

---
 Jenkinsfile                           | 11 ++++++
 tests/scripts/task_python_frontend.sh | 29 +-------------
 tests/scripts/task_python_legacy.sh   | 55 +++++++++++++++++++++++++++
 3 files changed, 67 insertions(+), 28 deletions(-)
 create mode 100755 tests/scripts/task_python_legacy.sh

diff --git a/Jenkinsfile b/Jenkinsfile
index 74f060585781..416469563bc3 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -283,6 +283,17 @@ stage('Integration Test') {
       }
     }
   },
+  'legacy: GPU': {
+    node('GPU') {
+      ws('workspace/tvm/legacy-python-gpu') {
+        init_git()
+        unpack_lib('gpu', tvm_multilib)
+        timeout(time: max_time, unit: 'MINUTES') {
+          sh "${docker_run} ${ci_gpu} ./tests/scripts/task_python_legacy.sh"
+        }
+      }
+    }
+  },
   'docs: GPU': {
     node('GPU') {
       ws('workspace/tvm/docs-python-gpu') {
diff --git a/tests/scripts/task_python_frontend.sh b/tests/scripts/task_python_frontend.sh
index 609b00149bad..9985d4ab7821 100755
--- a/tests/scripts/task_python_frontend.sh
+++ b/tests/scripts/task_python_frontend.sh
@@ -29,33 +29,6 @@ make cython3
 echo "Running relay TFLite frontend test..."
 python3 -m nose -v tests/python/frontend/tflite
 
-echo "Running nnvm unittest..."
-python3 -m nose -v nnvm/tests/python/unittest
-
-echo "Running nnvm compiler test..."
-python3 -m nose -v nnvm/tests/python/compiler
-
-echo "Running nnvm ONNX frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/onnx
-
-echo "Running nnvm MXNet frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/mxnet
-
-echo "Running nnvm Keras frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/keras
-
-echo "Running nnvm Tensorflow frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/tensorflow
-
-echo "Running nnvm CoreML frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/coreml
-
-echo "Running nnvm Caffe2 frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/caffe2
-
-echo "Running nnvm DarkNet frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/darknet || exit -1
-
 echo "Running relay MXNet frontend test..."
 python3 -m nose -v tests/python/frontend/mxnet
 
@@ -78,4 +51,4 @@ echo "Running relay caffe2 frontend test..."
 python3 -m nose -v tests/python/frontend/caffe2
 
 echo "Running relay DarkNet frontend test..."
-python3 -m nose -v tests/python/frontend/darknet || exit -1
+python3 -m nose -v tests/python/frontend/darknet
diff --git a/tests/scripts/task_python_legacy.sh b/tests/scripts/task_python_legacy.sh
new file mode 100755
index 000000000000..df1615bb8550
--- /dev/null
+++ b/tests/scripts/task_python_legacy.sh
@@ -0,0 +1,55 @@
+#!/bin/bash
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# Test cases for legacy code, will be deprecated in the future.
+set -e
+set -u
+
+export PYTHONPATH=nnvm/python:python:topi/python
+export OMP_NUM_THREADS=1
+
+# Rebuild cython
+make cython3
+
+echo "Running nnvm unittest..."
+python3 -m nose -v nnvm/tests/python/unittest
+
+
+echo "Running nnvm compiler test..."
+python3 -m nose -v nnvm/tests/python/compiler
+
+echo "Running nnvm ONNX frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/onnx
+
+echo "Running nnvm MXNet frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/mxnet
+
+echo "Running nnvm DarkNet frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/darknet
+
+echo "Running nnvm Keras frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/keras
+
+echo "Running nnvm Tensorflow frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/tensorflow
+
+echo "Running nnvm CoreML frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/coreml
+
+echo "Running nnvm Caffe2 frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/caffe2