Skip to content

[Grug] Add DeepEP profiling named scopes#6175

Merged
dlwh merged 1 commit into
mainfrom
codex/6139-deepep-trace-scopes
Jun 4, 2026
Merged

[Grug] Add DeepEP profiling named scopes#6175
dlwh merged 1 commit into
mainfrom
codex/6139-deepep-trace-scopes

Conversation

@dlwh

@dlwh dlwh commented Jun 4, 2026

Copy link
Copy Markdown
Member

Add named scopes around DeepEP layout, dispatch/combine transport, local packing, and collapse paths so profiler traces can attribute movement and local work separately.

Part of #6139

@dlwh dlwh added the agent-generated Created by automation/agent label Jun 4, 2026
@claude

claude Bot commented Jun 4, 2026

Copy link
Copy Markdown
Contributor

Claude finished @dlwh's task in 3m 41s —— View job


Code review

No issues found. Checked for bugs and CLAUDE.md/AGENTS.md compliance.

This PR is purely additive profiling instrumentation — with jax.named_scope(...) wrappers around existing logic in lib/levanter/src/levanter/grug/_moe/ep_deepep.py. No control-flow, signature, or import changes; all reindentation lines up; import jax is already present; scope names follow the project's snake_case convention.

  • Check PR status (closed/draft/prior review)
  • Identify relevant CLAUDE.md/AGENTS.md files
  • Summarize PR changes
  • Run 4 parallel review agents (CLAUDE.md + bugs)
  • Validate flagged issues
  • Emit stats event (stats endpoint blocked by sandbox; skill marks as best-effort)
  • Post summary comment
    Branch

@dlwh dlwh merged commit 22dfe05 into main Jun 4, 2026
32 checks passed
@dlwh dlwh deleted the codex/6139-deepep-trace-scopes branch June 4, 2026 22:28
@dlwh dlwh restored the codex/6139-deepep-trace-scopes branch June 4, 2026 22:28
@dlwh dlwh deleted the codex/6139-deepep-trace-scopes branch June 4, 2026 22:28
@dlwh dlwh restored the codex/6139-deepep-trace-scopes branch June 6, 2026 05:59
@dlwh dlwh deleted the codex/6139-deepep-trace-scopes branch June 13, 2026 05:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

agent-generated Created by automation/agent

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant