Skip to content

Rollback PR #37936: since it breaks ragged_collective_test_gpu in JAX when machine have >2 GPUs. #67594

Rollback PR #37936: since it breaks ragged_collective_test_gpu in JAX when machine have >2 GPUs.

Rollback PR #37936: since it breaks ragged_collective_test_gpu in JAX when machine have >2 GPUs. #67594

Triggered via pull request April 17, 2026 09:56
Status Success
Total duration 12m 34s
Artifacts 4

presubmit_benchmark.yml

on: pull_request
Generate Presubmit Matrix  /  Generate Matrix (PRESUBMIT)
2m 7s
Generate Presubmit Matrix / Generate Matrix (PRESUBMIT)
Matrix: run_benchmarks
Fit to window
Zoom out
Zoom in

Annotations

12 warnings
Run Benchmark (gemma3_1b_flax_call_x86_1h1d_presubmit)
Skipping unsupported unit: 3891 compute
Run Benchmark (gemma3_1b_flax_call_x86_1h1d_presubmit)
Skipping unsupported unit: 3891 compute
Run Benchmark (gemma3_1b_flax_call_x86_1h1d_presubmit)
Skipping unsupported unit: log messages before absl::InitializeLog() is called are written to STDERR
Run Benchmark (gemma2_2b_keras_jax_x86_1h1d_presubmit)
Skipping unsupported unit: 3931 compute
Run Benchmark (gemma2_2b_keras_jax_x86_1h1d_presubmit)
Skipping unsupported unit: 3931 compute
Run Benchmark (gemma2_2b_keras_jax_x86_1h1d_presubmit)
Skipping unsupported unit: log messages before absl::InitializeLog() is called are written to STDERR
Run Benchmark (gemma3_1b_flax_call_l4_1h1d_presubmit)
Skipping unsupported unit: 2683 compute
Run Benchmark (gemma3_1b_flax_call_l4_1h1d_presubmit)
Skipping unsupported unit: 2683 compute
Run Benchmark (gemma3_1b_flax_call_l4_1h1d_presubmit)
Skipping unsupported unit: log messages before absl::InitializeLog() is called are written to STDERR
Run Benchmark (gemma2_2b_keras_jax_l4_1h1d_presubmit)
Skipping unsupported unit: 2829 compute
Run Benchmark (gemma2_2b_keras_jax_l4_1h1d_presubmit)
Skipping unsupported unit: 2829 compute
Run Benchmark (gemma2_2b_keras_jax_l4_1h1d_presubmit)
Skipping unsupported unit: log messages before absl::InitializeLog() is called are written to STDERR

Artifacts

Produced during runtime
Name Size Digest
results-gemma2_2b_keras_jax_l4_1h1d_presubmit Expired
1.22 MB
sha256:6ce777c7b830093cfab0a053b62c92b477b59fab9b172bfc2511e98824940e92
results-gemma2_2b_keras_jax_x86_1h1d_presubmit Expired
2.62 MB
sha256:a898039e3ccdf1f586ab6c850778e1fc71296ba3945c31a4387576897ad7f6c6
results-gemma3_1b_flax_call_l4_1h1d_presubmit Expired
480 KB
sha256:5b62a1cf25d2657e93cc989e687d34ba1a43bf268fcdd60ec806820fdbf45110
results-gemma3_1b_flax_call_x86_1h1d_presubmit Expired
897 KB
sha256:ceee97c390a6eb23161ce7c4ac792f6f2c17bb693e1c3ba84089831a8a0e1b2e