Skip to content
Presubmit Benchmarks

[TPU/Raiden/StagingHostRam] Add num_staging_layers constructor parameter to PyTorch KVCacheManager, replacing TPU_RAIDEN_NUM_STAGING_LAYERS #2690

Sign in to view logs

[TPU/Raiden/StagingHostRam] Add num_staging_layers constructor parameter to PyTorch KVCacheManager, replacing TPU_RAIDEN_NUM_STAGING_LAYERS

[TPU/Raiden/StagingHostRam] Add num_staging_layers constructor parameter to PyTorch KVCacheManager, replacing TPU_RAIDEN_NUM_STAGING_LAYERS #2690

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Run Single Host Benchmarks / Validate registry security
1m 1s
Run Single Host Benchmarks / Generate matrix
2m 6s
Run Single Host Benchmarks / Run benchmark (h2d_d2h_gating_torch_tpu-v5e-single-node)
6m 9s
Run Single Host Benchmarks / Run benchmark (h2d_d2h_gating_jax_tpu-v5e-single-node)
3m 31s
Run Single Host Benchmarks / Run benchmark (h2h_cpp_gate_tpu-v5e-single-node)
3m 48s
Run Single Host Benchmarks / Analyze results
53s
Run Single Host Benchmarks / Bundle artifacts
1m 12s
18m 40s