Skip to content

Speed up CI: narrow OOB test and fix warp kernel cache#1326

Merged
erikfrey merged 2 commits into
google-deepmind:mainfrom
erikfrey:slim_bounds_check_ci
May 1, 2026
Merged

Speed up CI: narrow OOB test and fix warp kernel cache#1326
erikfrey merged 2 commits into
google-deepmind:mainfrom
erikfrey:slim_bounds_check_ci

Conversation

@erikfrey

@erikfrey erikfrey commented May 1, 2026

Copy link
Copy Markdown
Collaborator

Summary

Two CI speedups for the stable dependency test matrix.

Narrow out-of-bounds check

The --debug_mode out-of-bounds check was running all of io_test, but only test_model_batched_fields needs it. Narrow the pytest invocation to just that test.

Fix warp kernel cache

Add dep-type to the warp kernel cache key so that stable and locked uv dependency tests each get their own kernel cache. Previously both shared the same cache key (keyed on uv.lock), which is irrelevant for the stable job (pip install .). When Warp versions differed between locked and stable, the shared cache contained incompatible kernels, forcing recompilation.

@erikfrey erikfrey requested a review from thowell May 1, 2026 22:47
Add dep-type to the CI warp cache key so that stable and locked uv
dependency tests each get their own kernel cache. Previously both
shared the same cache keyed on uv.lock, which is irrelevant for
stable (pip install), causing cache misses when warp versions differ.
@erikfrey erikfrey changed the title Narrow out-of-bounds test to speed up CI Fix warp kernel cache for stable dependency CI tests May 1, 2026
@erikfrey erikfrey changed the title Fix warp kernel cache for stable dependency CI tests Speed up CI: narrow OOB test and fix warp kernel cache May 1, 2026
@erikfrey erikfrey merged commit 0093b8e into google-deepmind:main May 1, 2026
34 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants