Run GPU unit tests on T4x2

This will verify that the unit tests pass on both GPU types.

http://b/254301311

Author: djherbis

Jenkinsfile

@@ -126,17 +126,37 @@ pipeline {
               }
             }
             stage('Test GPU Image') {
-              options {
-                timeout(time: 20, unit: 'MINUTES')
-              }
-              steps {
-                sh '''#!/bin/bash
-                  set -exo pipefail
+              parallel {
+                stage('Test on P100') {
+                  agent { label 'ephemeral-linux-gpu' }
+                  options {
+                    timeout(time: 20, unit: 'MINUTES')
+                  }
+                  steps {
+                    sh '''#!/bin/bash
+                      set -exo pipefail
 
-                  date
-                  docker pull gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
-                  ./test --gpu --image gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
-                '''
+                      date
+                      docker pull gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
+                      ./test --gpu --image gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
+                    '''
+                  }
+                }
+                stage('Test on T4x2') {
+                  agent { label 'ephemeral-linux-gpu-t4x2' }
+                  options {
+                    timeout(time: 20, unit: 'MINUTES')
+                  }
+                  steps {
+                    sh '''#!/bin/bash
+                      set -exo pipefail
+
+                      date
+                      docker pull gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
+                      ./test --gpu --image gcr.io/kaggle-private-byod/python:${PRETEST_TAG}
+                    '''
+                  }
+                }
               }
             }
             stage('Diff GPU Image') {