Skip to content

Commit 6a149fc

Browse files
committed
Run GPU unit tests on T4x2
This will verify that the unit tests pass on both GPU types. http://b/254301311
1 parent 1517713 commit 6a149fc

File tree

1 file changed

+30
-10
lines changed

1 file changed

+30
-10
lines changed

Jenkinsfile

Lines changed: 30 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -126,17 +126,37 @@ pipeline {
126126
}
127127
}
128128
stage('Test GPU Image') {
129-
options {
130-
timeout(time: 20, unit: 'MINUTES')
131-
}
132-
steps {
133-
sh '''#!/bin/bash
134-
set -exo pipefail
129+
parallel {
130+
stage('Test on P100') {
131+
agent { label 'ephemeral-linux-gpu' }
132+
options {
133+
timeout(time: 20, unit: 'MINUTES')
134+
}
135+
steps {
136+
sh '''#!/bin/bash
137+
set -exo pipefail
135138
136-
date
137-
docker pull gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
138-
./test --gpu --image gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
139-
'''
139+
date
140+
docker pull gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
141+
./test --gpu --image gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
142+
'''
143+
}
144+
}
145+
stage('Test on T4x2') {
146+
agent { label 'ephemeral-linux-gpu-t4x2' }
147+
options {
148+
timeout(time: 20, unit: 'MINUTES')
149+
}
150+
steps {
151+
sh '''#!/bin/bash
152+
set -exo pipefail
153+
154+
date
155+
docker pull gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
156+
./test --gpu --image gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
157+
'''
158+
}
159+
}
140160
}
141161
}
142162
stage('Diff GPU Image') {

0 commit comments

Comments
 (0)