From ac23f89e717df858b572fdc8b1033a9ab0f196c5 Mon Sep 17 00:00:00 2001
From: Jonathan Calderon Chavez <chavezcalderon@google.com>
Date: Tue, 4 Jun 2024 23:11:44 +0000
Subject: [PATCH 1/2] par

---
 Dockerfile.tmpl | 2 +-
 Jenkinsfile     | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 2b4cd33d..3acca712 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -266,7 +266,7 @@ RUN pip install opencv-contrib-python opencv-python && \
     /tmp/clean-layer.sh
 
 # Pin scipy until we update JAX b/335003097
-RUN pip install "scipy==1.12.0" \
+RUN pip install scipy \
         # Scikit-learn accelerated library for x86
         "scikit-learn-intelex>=2023.0.1" \
         # HDF5 support
diff --git a/Jenkinsfile b/Jenkinsfile
index 5137c675..23b23912 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -153,6 +153,7 @@ pipeline {
             }
             stage('Test GPU Image') {
               stages {
+                parallel {
                 stage('Test on P100') {
                   agent { label 'ephemeral-linux-gpu' }
                   options {
@@ -187,6 +188,7 @@ pipeline {
                     }
                   }
                 }
+                }
               }
             }
             stage('Diff GPU Image') {

From bfbb6eb1cb88f80067b8918cea6041f0bfcfa14f Mon Sep 17 00:00:00 2001
From: Jonathan Calderon Chavez <chavezcalderon@google.com>
Date: Tue, 4 Jun 2024 23:21:49 +0000
Subject: [PATCH 2/2] par2

---
 Jenkinsfile | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index 23b23912..aafb17f7 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -152,8 +152,7 @@ pipeline {
               }
             }
             stage('Test GPU Image') {
-              stages {
-                parallel {
+              parallel {
                 stage('Test on P100') {
                   agent { label 'ephemeral-linux-gpu' }
                   options {
@@ -188,7 +187,6 @@ pipeline {
                     }
                   }
                 }
-                }
               }
             }
             stage('Diff GPU Image') {