Skip to content

Improve scheduler dispatch, fairness, and reliability#2208

Open
rejuvenile wants to merge 4 commits intoTraceMachina:mainfrom
rejuvenile:pr/04-scheduler-dispatch
Open

Improve scheduler dispatch, fairness, and reliability#2208
rejuvenile wants to merge 4 commits intoTraceMachina:mainfrom
rejuvenile:pr/04-scheduler-dispatch

Conversation

@rejuvenile
Copy link

@rejuvenile rejuvenile commented Mar 12, 2026

Summary

  • Improve action dispatch fairness and worker selection
  • Add load-aware scheduling with worker capacity tracking
  • Add locality-based scheduling (cache-aware worker selection)
  • Tree resolution caching for scheduler dispatch

Test plan

  • cargo check passes
  • CI checks pass

Stack: 4/11 — depends on #2207

🤖 Generated with Claude Code


This change is Reviewable

rejuvenile and others added 4 commits March 11, 2026 16:54
Update tonic, prost, and other dependencies to latest versions.
Regenerate protobuf bindings. Add aws-lc-rs and rayon support
to digest hasher. Update BUILD.bazel files for LRE toolchain.

Co-Authored-By: Claude <noreply@anthropic.com>
Rewrite existence cache to prevent stale positives by bypassing cache
on update and cleaning entries on NotFound. Fix BatchUpdateBlobs
duplicate digest handling. Add POSIX_FADV_SEQUENTIAL for read-ahead.
Pre-set CAS files to 0o555 to avoid redundant chmod on hardlink.
Fix LRU eviction ordering at startup by sorting files by atime.
Add stall detector for store operations. Replace async Mutex with
parking_lot in EvictingMap. Increase gRPC connections_per_endpoint
default to 32.

Co-Authored-By: Claude <noreply@anthropic.com>
Fix ByteStream protocol compliance for large blobs. Add gRPC error
details with proper status codes. Add max_total_batch_size config
for BatchReadBlobs/BatchUpdateBlobs. Improve capabilities server
to report actual supported features. Add TLS improvements and
server startup configuration options. Downgrade per-request
transfer logs to debug level.

Co-Authored-By: Claude <noreply@anthropic.com>
Add load-aware worker selection with CPU load tracking. Implement
directory cache and subtree coverage scoring for locality-aware
scheduling. Add stall detection for queued actions. Support
SIGKILL retry on worker timeout. Handle FAILED_PRECONDITION for
missing inputs. Add fair round-robin dispatch via LRU promotion.
Downgrade per-action dispatch logs to debug level.

Co-Authored-By: Claude <noreply@anthropic.com>
@CLAassistant
Copy link

CLAassistant commented Mar 12, 2026

CLA assistant check
All committers have signed the CLA.

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