Skip to content

Rebase Arctic Inference onto vLLM v0.26.0 - #269

Open
sfc-gh-mhidayetoglu wants to merge 12 commits into
snowflakedb:mainfrom
sfc-gh-mhidayetoglu:rebase/vllm_v26
Open

Rebase Arctic Inference onto vLLM v0.26.0#269
sfc-gh-mhidayetoglu wants to merge 12 commits into
snowflakedb:mainfrom
sfc-gh-mhidayetoglu:rebase/vllm_v26

Conversation

@sfc-gh-mhidayetoglu

@sfc-gh-mhidayetoglu sfc-gh-mhidayetoglu commented Aug 13, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Rebase Arctic Inference from vLLM v0.18.0 → v0.24.0 → v0.26.0, including CUDA graph fixes.
  • Update version pins in pyproject.toml and adjust plugin/patches/model runner code for the newer vLLM APIs.
  • Make the version gate strict: with ARCTIC_INFERENCE_ENABLED=1, a vLLM version mismatch now raises instead of silently running unpatched (ARCTIC_INFERENCE_SKIP_VERSION_CHECK=1 bypasses).
  • Fold the rollout-replay (max_tokens_n) feature into the plugin patching system and drop the standalone source-patch scripts.
  • Add rebase/architecture/versioning skills and supporting scripts under skills/.

Changes

  • Core vLLM integration updates: plugin.py, patches.py, model_runner.py, args.py, ulysses.py, attention forest cascade, spec-dec fp8, SwiftKV llama.
  • Versioning: plugin.py hard-fails on enabled+mismatch; utils.py introduces VLLM_PATCH_VERSION as the single source of truth (decoupled from install extras); server/worker.py engine-args selection simplified accordingly.
  • Rollout replay: new arctic_inference/vllm/sampling.py (ParentRequestPatch) applied in apply_arctic_patches(); per-child lengths passed via SamplingParams.extra_args["max_tokens_n"]. Removes benchmark/rollout/{parallel_sampling.patch,sampling_params.patch,patch_sampling.sh}; updates benchmark/rollout/README.md.
  • Build/deps: pyproject.toml, csrc/custom_ops/CMakeLists.txt.
  • Docs/skills: skills/arctic-inference-architecture, skills/arctic-vllm-versioning, skills/rebase-arctic-inference (+ example rebase plans/reports and helper scripts).
  • Tests: tests/unit_tests/test_custom_ops.py.

Comment thread arctic_inference/vllm/plugin.py
Comment thread pyproject.toml Outdated
Comment thread pyproject.toml Outdated
Comment thread pyproject.toml Outdated
Comment thread skills/arctic-vllm-versioning/SKILL.md Outdated
Comment thread skills/arctic-vllm-versioning/reference.md
@sfc-gh-sbekman
sfc-gh-sbekman self-requested a review August 24, 2026 23:07

@sfc-gh-sbekman sfc-gh-sbekman left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you, Mert

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants