Skip to content

feat(attention): add SBHD DSv4 hybrid orchestration - #5948

Closed
FDecaYed wants to merge 1 commit into
NVIDIA:pull-request/5947from
FDecaYed:codex/dsv4-hybrid-attention
Closed

feat(attention): add SBHD DSv4 hybrid orchestration#5948
FDecaYed wants to merge 1 commit into
NVIDIA:pull-request/5947from
FDecaYed:codex/dsv4-hybrid-attention

Conversation

@FDecaYed

Copy link
Copy Markdown
Contributor

What

Enable the SBHD-only DSv4 Hybrid attention variant:

  • compose local sliding-window attention with compressed sparse attention;
  • add grouped low-rank output projection;
  • validate the model variant, attention ratios, and unsupported modes;
  • add a backend-explicit module-spec boundary without coupling to GPTModel or models.backends;
  • propagate MTP layer identity through existing attention constructors;
  • cover construction, QKV, forward/backward, grouped output, RoPE, config, and backend providers.

This PR owns orchestration only. Its Files changed view does not repeat the RoPE or CSA implementations from its parents.

Stack

This PR is stacked on #5947 and transitively #5944.

#5942 is an independent semantic merge prerequisite for correct end-to-end indexer-loss scaling, but is not a Git-base dependency of this slice.

Provenance

Deliberately excluded

  • GPTModel and HybridModel wiring;
  • fused CSA/DSA kernels;
  • packed-sequence/THD and context-parallel support (CP is rejected early);
  • HCA, Hash MoE, mHC, and FP8 marker integration.

Testing

  • isort, ruff, Black, Python compilation, and git diff --check passed for the exact slice.
  • Focused pytest collection is blocked locally because this host lacks Torch/CUDA; PR CI owns execution.

Enable SBHD-only DSv4 Hybrid attention orchestration, model-variant validation, backend-explicit specs, and MTP layer-ratio plumbing.

Reconstructed from NVIDIA#4458, NVIDIA#4518, NVIDIA#5018, and NVIDIA#5526 as part of the frozen feature set in NVIDIA#5795.

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 24d4d07

Copy link
Copy Markdown
Contributor Author

Closing this draft until Round 2. Its logical successor is D2 (SBHD DSv4 orchestration and indexer-loss scaling), rebuilt after D1 and P6 merge and absorbing #5942. 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