Skip to content

[Fix] Scope FSDP overlap carry to the leftover compute node so earlier all-gathers still hoist - #73

Merged
jiahy0825 merged 1 commit into
SandAI-org:mainfrom
wtr0504:dev/reorder
Sep 10, 2026
Merged

[Fix] Scope FSDP overlap carry to the leftover compute node so earlier all-gathers still hoist#73
jiahy0825 merged 1 commit into
SandAI-org:mainfrom
wtr0504:dev/reorder

Conversation

@wtr0504

@wtr0504 wtr0504 commented Sep 10, 2026

Copy link
Copy Markdown
Collaborator

🗂️ PR Category

  • ✨ New Feature
  • 🚀 Optimization (performance, memory, etc.)
  • 💥 Breaking Change
  • 🐛 Bug Fix
  • 🛠️ Development / Refactoring
  • 📚 Documentation
  • 🧹 Chore (Dependencies, CI/CD, Configuration, etc.)
  • 🧪 Testing

📝 Description

  • Keep whole-node carry so a 95µs collective does not spend a 33ms attention.
  • Track carry_idx as the node holding unused leftover. When the next gather already sits upstream (cur < carry_idx), drop carry and walk a fresh window.
  • Unscoped carry treated a later layer's leftover as a hide for earlier launches, so large weight all-gathers stayed at cur and the wait was exposed.

@wtr0504 wtr0504 changed the title [Fix] Scope FSDP overlap carry to the leftover compute node so earlie… [Fix] Scope FSDP overlap carry to the leftover compute node so earlier all-gathers still hoist Sep 10, 2026
@wtr0504 wtr0504 added the ci:run Trigger CI integration tests label Sep 10, 2026
@github-actions github-actions Bot removed the ci:run Trigger CI integration tests label Sep 10, 2026

@jiahy0825 jiahy0825 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@jiahy0825
jiahy0825 merged commit 5166b8e into SandAI-org:main Sep 10, 2026
23 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