Skip to content

fix(configs): pin head unschedulable — train (batch 4)#911

Merged
Aydin-ab merged 1 commit into
mainfrom
fix/batch-4-head-pin-deps
Jul 22, 2026
Merged

fix(configs): pin head unschedulable — train (batch 4)#911
Aydin-ab merged 1 commit into
mainfrom
fix/batch-4-head-pin-deps

Conversation

@Aydin-ab

@Aydin-ab Aydin-ab commented Jul 18, 2026

Copy link
Copy Markdown
Contributor

Pins the head unschedulable (CPU: 0) for the batch-4 train templates. deepspeed_finetune, pytorch-fsdp, ray_train_workloads have CPU-only heads. distributing-pytorch's head was a GPU instance only so its Step-1 single-machine baseline could use a local GPU — switch it to a CPU-only coordinator and run Step 1 on a GPU worker via a Ray task.

Testing

/test-template on all four (distributing-pytorch's notebook refactor validated on GPU CI, buildkite #533 as #922 + re-run here).

Notes

distributing-pytorch was briefly standalone (#922, now closed); folded back to keep batch 4 in one PR.

https://claude.ai/code/session_01QmLc4yWtmC3PX2NwWzPbzD

@Aydin-ab

Copy link
Copy Markdown
Contributor Author

/test-template deepspeed_finetune pytorch-fsdp ray_train_workloads

@Aydin-ab

Copy link
Copy Markdown
Contributor Author

/test-template distributing-pytorch

@Aydin-ab
Aydin-ab force-pushed the fix/batch-4-head-pin-deps branch 2 times, most recently from 6e57695 to 5af157c Compare July 20, 2026 22:53
deepspeed_finetune, pytorch-fsdp, ray_train_workloads have CPU-only heads — pin
head_node.resources CPU: 0 so work runs on the GPU worker groups. distributing-pytorch's
head was a GPU instance only so the Step-1 single-machine baseline could use a local GPU;
switch it to a CPU-only coordinator (m5.2xlarge / n2-standard-8, CPU: 0) and run Step 1 on
a GPU worker via a Ray task.

Claude-Session: https://claude.ai/code/session_01QmLc4yWtmC3PX2NwWzPbzD
Signed-off-by: Aydin Abiar <aydin@anyscale.com>
@Aydin-ab
Aydin-ab force-pushed the fix/batch-4-head-pin-deps branch from 5af157c to 472dd15 Compare July 22, 2026 00:29
@Aydin-ab

Copy link
Copy Markdown
Contributor Author

/test-template distributing-pytorch

@Aydin-ab
Aydin-ab merged commit f7ad867 into main Jul 22, 2026
5 checks passed
@Aydin-ab
Aydin-ab deleted the fix/batch-4-head-pin-deps branch July 22, 2026 22:15
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