Increase Kokoro GPU timeout to 240 minutes for JAX, TensorFlow, and PyTorch (keras-team#2801)

laxmareddyp · web-flow · commit 4f2f6c430d52 · 2026-07-14T17:31:31.000-07:00
diff --git a/.kokoro/github/ubuntu/gpu/jax/continuous.cfg b/.kokoro/github/ubuntu/gpu/jax/continuous.cfg
@@ -30,5 +30,5 @@ before_action {
   }
 }
 
-# Set timeout lower from default 180 mins
-timeout_mins: 180
+# Set timeout to 240 mins
+timeout_mins: 240
diff --git a/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg b/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg
@@ -30,5 +30,5 @@ before_action {
   }
 }
 
-# Set timeout to 180 mins
-timeout_mins: 180
+# Set timeout to 240 mins
+timeout_mins: 240
diff --git a/.kokoro/github/ubuntu/gpu/tensorflow/continuous.cfg b/.kokoro/github/ubuntu/gpu/tensorflow/continuous.cfg
@@ -30,5 +30,5 @@ before_action {
   }
 }
 
-# Set timeout lower from default 180 mins
-timeout_mins: 180
+# Set timeout to 240 mins
+timeout_mins: 240
diff --git a/.kokoro/github/ubuntu/gpu/tensorflow/presubmit.cfg b/.kokoro/github/ubuntu/gpu/tensorflow/presubmit.cfg
@@ -30,5 +30,5 @@ before_action {
   }
 }
 
-# Set timeout to 180 mins
-timeout_mins: 180
+# Set timeout to 240 mins
+timeout_mins: 240
diff --git a/.kokoro/github/ubuntu/gpu/torch/continuous.cfg b/.kokoro/github/ubuntu/gpu/torch/continuous.cfg
@@ -30,5 +30,5 @@ before_action {
   }
 }
 
-# Set timeout to 120 mins from default 180 mins
-timeout_mins: 180
+# Set timeout to 240 mins
+timeout_mins: 240
diff --git a/.kokoro/github/ubuntu/gpu/torch/presubmit.cfg b/.kokoro/github/ubuntu/gpu/torch/presubmit.cfg
@@ -30,5 +30,5 @@ before_action {
   }
 }
 
-# Set timeout to 180 mins
-timeout_mins: 180
+# Set timeout to 240 mins
+timeout_mins: 240

Original file line number	Diff line number	Diff line change
`@@ -30,5 +30,5 @@ before_action {`
`30`	`30`	`}`
`31`	`31`	`}`
`32`	`32`
`33`		`-# Set timeout lower from default 180 mins`
`34`		`-timeout_mins: 180`
	`33`	`+# Set timeout to 240 mins`
	`34`	`+timeout_mins: 240`