update SP+EP - #209
Open
sfc-gh-mhidayetoglu wants to merge 86 commits into
Open
Conversation
…ffix_draft_token_ids` (#195)
Co-authored-by: Aurick Qiao <aurick.qiao@snowflake.com>
Shift Parallelism paper link: https://arxiv.org/abs/2509.16495
This reverts commit 170082f.
Co-authored-by: Mert Hidayetoglu <mert.hidayetoglu@snowflake.com> Co-authored-by: Aurick Qiao <aurick.qiao@snowflake.com>
Co-authored-by: Ye Wang <ye.wang@snowflake.com>
sfc-gh-mhidayetoglu
requested review from
sfc-gh-aqiao,
sfc-gh-goliaro,
sfc-gh-jrasley and
sfc-gh-yewang
as code owners
October 15, 2025 00:31
Co-authored-by: YanlinDu <yanlin.du@snowflake.com> Co-authored-by: Cursor <cursoragent@cursor.com>
Signed-off-by: Stas Bekman <stas.bekman@snowflake.com>
Signed-off-by: Stas Bekman <stas.bekman@snowflake.com>
…-eng/ArcticInference-internal into tunji/verl_integration
… (#76) Co-authored-by: Ye Wang <ye.wang@snowflake.com> Co-authored-by: Jeff Rasley <jeff.rasley@snowflake.com> Co-authored-by: 151130f5470be3da5eea21fd02baa3_snow <bb9ecf6d788c5c0e03a13a6ec00039@snowflake.com> Co-authored-by: Michael Wyatt <michael.wyatt@snowflake.com> Co-authored-by: Cursor <cursoragent@cursor.com>
Compare sender HF names against the receiver model's expected set and fail loud on mismatch so Qwen3-32B colocated runs catch architecture drift early. Co-authored-by: Cursor <cursoragent@cursor.com>
- receiver: raise on incomplete CUDA IPC load (tensor-count mismatch), validate received parameter names on NCCL/CPU paths, and accept either serialized bytes or (name, tensor) pairs in load_weights_from_cpu - replica_pool: raise RuntimeError when any worker's sync_weights status is not "done" instead of swallowing the failure - api/multi_model: add /spec_weights_info and /sync_spec_weights endpoints plus Driver forwarding for multi-model mode Co-authored-by: Cursor <cursoragent@cursor.com>
Add load_weights_cuda_ipc_chunk to the weight-sync receiver, which loads one param (or small chunk) at a time from CUDA IPC handles instead of requiring the whole model in one payload. It raises on any GPU missing a handle and runs full param-name validation on the final chunk (context=cuda_ipc_stream). Add the matching InferenceWorker collective_rpc wrapper. This bounds peak extra GPU memory during sync to one full param per GPU. Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
Signed-off-by: Stas Bekman <stas.bekman@snowflake.com>
…Arctic Platform (#263)
Arctic rl/integration
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
13 participants
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.
No description provided.