@@ -561,7 +561,7 @@ def test_unix_python2_cpu() {
561561
562562def 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
596596def 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
646646def 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
748748def 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
11091109def 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