Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 007ae76

Browse files
committed
Updates cuDNN enabled gpu tests to run on P3 instances
1 parent 29578b5 commit 007ae76

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

ci/jenkins/Jenkins_steps.groovy

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -561,7 +561,7 @@ def test_unix_python2_cpu() {
561561

562562
def test_unix_python2_gpu() {
563563
return ['Python2: GPU': {
564-
node(NODE_LINUX_GPU) {
564+
node(NODE_LINUX_GPU_P3) {
565565
ws('workspace/ut-python2-gpu') {
566566
try {
567567
utils.unpack_and_init('gpu', mx_lib, true)
@@ -595,7 +595,7 @@ def test_unix_python2_quantize_gpu() {
595595

596596
def test_unix_python2_mkldnn_gpu() {
597597
return ['Python2: MKLDNN-GPU': {
598-
node(NODE_LINUX_GPU) {
598+
node(NODE_LINUX_GPU_P3) {
599599
ws('workspace/ut-python2-mkldnn-gpu') {
600600
try {
601601
utils.unpack_and_init('mkldnn_gpu', mx_mkldnn_lib, true)
@@ -645,7 +645,7 @@ def test_unix_python3_mkl_cpu() {
645645

646646
def test_unix_python3_gpu() {
647647
return ['Python3: GPU': {
648-
node(NODE_LINUX_GPU) {
648+
node(NODE_LINUX_GPU_P3) {
649649
ws('workspace/ut-python3-gpu') {
650650
try {
651651
utils.unpack_and_init('gpu', mx_lib, true)
@@ -747,7 +747,7 @@ def test_unix_python3_mkldnn_mkl_cpu() {
747747

748748
def test_unix_python3_mkldnn_gpu() {
749749
return ['Python3: MKLDNN-GPU': {
750-
node(NODE_LINUX_GPU) {
750+
node(NODE_LINUX_GPU_P3) {
751751
ws('workspace/ut-python3-mkldnn-gpu') {
752752
try {
753753
utils.unpack_and_init('mkldnn_gpu', mx_mkldnn_lib, true)
@@ -1108,7 +1108,7 @@ def test_centos7_python3_cpu() {
11081108

11091109
def test_centos7_python3_gpu() {
11101110
return ['Python3: CentOS 7 GPU': {
1111-
node(NODE_LINUX_GPU) {
1111+
node(NODE_LINUX_GPU_P3) {
11121112
ws('workspace/build-centos7-gpu') {
11131113
timeout(time: max_time, unit: 'MINUTES') {
11141114
try {

0 commit comments

Comments
 (0)