Repository navigation
[Refactor][TCPCG] Relocate shared graph tensor and DSA head-gate helpers (1/9) - #42285
Merged
Oasis-Git merged 3 commits intoOct 6, 2026
Merged
Conversation
Oasis-Git
marked this pull request as ready for review
October 2, 2026 23:17
Oasis-Git
requested review from
1am9trash,
Fridge003,
HaiShaw,
YAMY1234,
Ying1123,
hebiao064,
hnyls2002,
hubertlu-tw,
ispobock,
kkHuang-amd,
merrymercy and
rainj-me
as code owners
October 2, 2026 23:17
This was referenced Oct 2, 2026
Collaborator
Author
|
/rerun-failed-ci |
Collaborator
Author
|
/rerun-failed-ci |
Collaborator
Author
|
CI Pass with runner missing: https://github.com/sgl-project/sglang/actions/runs/37249466699/job/111936696722?pr=42285 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This is step 1/9, split from the original TCPCG removal PR #41634. It relocates shared helpers so BCG and torch.compile support can remain independent of the later TCPCG removal. Existing graph execution and TCPCG support remain in place at this step.
Series
Only step 1 is submitted upstream so far. Later steps will be submitted after review. This PR has no dependency on those later steps. The disputed relocation of inactive compiler/reference files is excluded from this series pending a separate decision.
Changes
weak_ref_tensor.pyfromcompilation/tomodel_executor/runner_backend_utils/and update both the BCG and TCPCG imports.attention/dsa/head_gate.py, retaining their CUDA/ROCm guards, custom-op registration, and fake implementations.The weak tensor helper is byte-identical to the original. All four relocated DSA helper implementations are AST-identical; only their surrounding module and comments change.
Validation
CI States
Latest PR Test (Base): ❌ Run #37249466699⚠️ Not enabled -- add
Latest PR Test (Extra):
run-ci-extralabel to opt in.Latest PR Test (AMD ROCm 10): ❌ Run #37249466729