Arctic rl/integration - #264
Merged
Merged
Conversation
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: jenkins-snowci-mergequeue-0-emu[bot] <232694648+jenkins-snowci-mergequeue-0-emu[bot]@users.noreply.github.com>
Co-authored-by: Sourcegraph <batch-changes@sourcegraph.com> Co-authored-by: EMU Migration Bot <emu-migration@snowflake.com>
Cherry-picked from public repo: #258 Co-authored-by: Xinyu Lian <lian7@illinois.edu> Made-with: Cursor
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
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)
sfc-gh-mhidayetoglu
requested review from
sfc-gh-goliaro,
sfc-gh-jrasley and
sfc-gh-reyazda
as code owners
June 24, 2026 17:51
sfc-gh-sbekman
approved these changes
Jun 24, 2026
sfc-gh-jrasley
approved these changes
Jun 24, 2026
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
8 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.