From 007ae76e37b84d359085ad8ef744a232f919b805 Mon Sep 17 00:00:00 2001 From: Per Goncalves da Silva Date: Mon, 1 Apr 2019 19:57:18 +0200 Subject: [PATCH] Updates cuDNN enabled gpu tests to run on P3 instances --- ci/jenkins/Jenkins_steps.groovy | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/ci/jenkins/Jenkins_steps.groovy b/ci/jenkins/Jenkins_steps.groovy index cebfe0413fce..c058e9e5e22f 100644 --- a/ci/jenkins/Jenkins_steps.groovy +++ b/ci/jenkins/Jenkins_steps.groovy @@ -561,7 +561,7 @@ def test_unix_python2_cpu() { def test_unix_python2_gpu() { return ['Python2: GPU': { - node(NODE_LINUX_GPU) { + node(NODE_LINUX_GPU_P3) { ws('workspace/ut-python2-gpu') { try { utils.unpack_and_init('gpu', mx_lib, true) @@ -595,7 +595,7 @@ def test_unix_python2_quantize_gpu() { def test_unix_python2_mkldnn_gpu() { return ['Python2: MKLDNN-GPU': { - node(NODE_LINUX_GPU) { + node(NODE_LINUX_GPU_P3) { ws('workspace/ut-python2-mkldnn-gpu') { try { utils.unpack_and_init('mkldnn_gpu', mx_mkldnn_lib, true) @@ -645,7 +645,7 @@ def test_unix_python3_mkl_cpu() { def test_unix_python3_gpu() { return ['Python3: GPU': { - node(NODE_LINUX_GPU) { + node(NODE_LINUX_GPU_P3) { ws('workspace/ut-python3-gpu') { try { utils.unpack_and_init('gpu', mx_lib, true) @@ -747,7 +747,7 @@ def test_unix_python3_mkldnn_mkl_cpu() { def test_unix_python3_mkldnn_gpu() { return ['Python3: MKLDNN-GPU': { - node(NODE_LINUX_GPU) { + node(NODE_LINUX_GPU_P3) { ws('workspace/ut-python3-mkldnn-gpu') { try { utils.unpack_and_init('mkldnn_gpu', mx_mkldnn_lib, true) @@ -1108,7 +1108,7 @@ def test_centos7_python3_cpu() { def test_centos7_python3_gpu() { return ['Python3: CentOS 7 GPU': { - node(NODE_LINUX_GPU) { + node(NODE_LINUX_GPU_P3) { ws('workspace/build-centos7-gpu') { timeout(time: max_time, unit: 'MINUTES') { try {