Skip to content

fix(pipeline): scale DSv4 hybrid indexer loss - #5942

Closed
FDecaYed wants to merge 1 commit into
NVIDIA:mainfrom
FDecaYed:codex/dsv4-loss-scale-hook
Closed

fix(pipeline): scale DSv4 hybrid indexer loss#5942
FDecaYed wants to merge 1 commit into
NVIDIA:mainfrom
FDecaYed:codex/dsv4-loss-scale-hook

Conversation

@FDecaYed

Copy link
Copy Markdown
Contributor

What

Teach the pipeline loss-scale selector that the dsv4_hybrid attention variant uses the same DSA indexer-loss auto-scaler as dsa. The parameterized test verifies both names and preserves the caller config.

This two-file correctness hook is deliberately independent: peeling it away keeps the attention implementation PR out of the pipeline-parallel CODEOWNER group.

Provenance

Testing

  • ruff, Black --check, Python compilation, and git diff --check passed.
  • Focused pytest execution is blocked because the repository uv environment on this host has no Torch; execution is left to CI.

Use the existing DSA indexer-loss auto-scaler for the DSv4 hybrid attention variant and cover both variant names.

Signed-off-by: Deyu Fu <Deyu.Foo@gmail.com>
@copy-pr-bot

copy-pr-bot Bot commented Jul 21, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@FDecaYed

Copy link
Copy Markdown
Contributor Author

/ok to test 1b08c42

Copy link
Copy Markdown
Contributor Author

Closing this thin standalone hook because the indexer-loss scheduling change is absorbed into DSv4 child D2 together with SBHD orchestration and its tests. Tracking PR: #5795.

@FDecaYed FDecaYed closed this Jul 23, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant