Commit 3ed70bb
authored
Merge pull request Kaggle#1189 from Kaggle/run-gpu-tests-t4
Run GPU unit tests on T4x21 file changed
Lines changed: 30 additions & 10 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -126,17 +126,37 @@ pipeline { | |||
| 126 | 126 | } | |
| 127 | 127 | } | |
| 128 | 128 | stage('Test GPU Image') { | |
| 129 | - options { | ||
| 130 | - timeout(time: 20, unit: 'MINUTES') | ||
| 131 | - } | ||
| 132 | - steps { | ||
| 133 | - sh '''#!/bin/bash | ||
| 134 | - set -exo pipefail | ||
| 129 | + stages { | ||
| 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 | ||
| 135 | 138 | ||
| 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 | + } | ||
| 140 | 160 | } | |
| 141 | 161 | } | |
| 142 | 162 | stage('Diff GPU Image') { | |
0 commit comments