From f6d3ceecbb5f93a17ad7d5934d770bd9346ff1b9 Mon Sep 17 00:00:00 2001
From: Tristan Konolige <tristan.konolige@gmail.com>
Date: Wed, 26 Aug 2020 19:29:24 -0700
Subject: [PATCH] [TESTS] add gpuonly tests for python unittests and
 integration (#6346)

---
 Jenkinsfile                                      |  4 ++--
 tests/scripts/task_python_integration_gpuonly.sh | 19 +++++++++++++++++++
 tests/scripts/task_python_unittest_gpuonly.sh    | 19 +++++++++++++++++++
 3 files changed, 40 insertions(+), 2 deletions(-)
 create mode 100755 tests/scripts/task_python_integration_gpuonly.sh
 create mode 100755 tests/scripts/task_python_unittest_gpuonly.sh

diff --git a/Jenkinsfile b/Jenkinsfile
index 7df0d3f..77b4e4b 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -202,8 +202,8 @@ stage('Unit Test') {
         unpack_lib('gpu', tvm_multilib)
         timeout(time: max_time, unit: 'MINUTES') {
           sh "${docker_run} ${ci_gpu} ./tests/scripts/task_sphinx_precheck.sh"
-          sh "${docker_run} ${ci_gpu} ./tests/scripts/task_python_unittest.sh"
-          sh "${docker_run} ${ci_gpu} ./tests/scripts/task_python_integration.sh"
+          sh "${docker_run} ${ci_gpu} ./tests/scripts/task_python_unittest_gpuonly.sh"
+          sh "${docker_run} ${ci_gpu} ./tests/scripts/task_python_integration_gpuonly.sh"
         }
       }
     }
diff --git a/tests/scripts/task_python_integration_gpuonly.sh b/tests/scripts/task_python_integration_gpuonly.sh
new file mode 100755
index 0000000..6b2755a
--- /dev/null
+++ b/tests/scripts/task_python_integration_gpuonly.sh
@@ -0,0 +1,19 @@
+#!/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.
+
+./tests/scripts/task_python_integration.sh
diff --git a/tests/scripts/task_python_unittest_gpuonly.sh b/tests/scripts/task_python_unittest_gpuonly.sh
new file mode 100755
index 0000000..637be67
--- /dev/null
+++ b/tests/scripts/task_python_unittest_gpuonly.sh
@@ -0,0 +1,19 @@
+#!/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.
+
+./tests/scripts/task_python_unittest.sh
-- 
2.7.4