Updates cuDNN enabled gpu tests to run on P3 instances

apache · Apr 1, 2019 · 007ae76 · 007ae76
1 parent 29578b5
commit 007ae76
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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 {