Increase JAX GPU tests timeout to 2 hours (#21915)

abheesht17 · web-flow · commit 9f7bd4091cb2 · 2025-12-10T09:01:57.000-08:00
diff --git a/.kokoro/github/ubuntu/gpu/jax/continuous.cfg b/.kokoro/github/ubuntu/gpu/jax/continuous.cfg
@@ -12,5 +12,5 @@ env_vars: {
    value: "jax"
 }
 
-# Set timeout to 60 mins from default 180 mins
-timeout_mins: 60
+# Set timeout to 120 mins from default 180 mins
+timeout_mins: 120
diff --git a/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg b/.kokoro/github/ubuntu/gpu/jax/presubmit.cfg
@@ -12,5 +12,5 @@ env_vars: {
    value: "jax"
 }
 
-# Set timeout to 60 mins from default 180 mins
-timeout_mins: 60
+# Set timeout to 120 mins from default 180 mins
+timeout_mins: 120

Original file line number	Diff line number	Diff line change
`@@ -12,5 +12,5 @@ env_vars: {`
`12`	`12`	`value: "jax"`
`13`	`13`	`}`
`14`	`14`
`15`		`-# Set timeout to 60 mins from default 180 mins`
`16`		`-timeout_mins: 60`
	`15`	`+# Set timeout to 120 mins from default 180 mins`
	`16`	`+timeout_mins: 120`