Merge branch 'main' into codex/rocm-artifact-tensorizer

Fix ROCm artifact test dependencies
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
2026-06-18 14:28:51 -05:00 · 2026-06-04 21:30:21 -05:00
342 changed files with 6372 additions and 10314 deletions
@@ -21,10 +21,6 @@ steps:
    timeout_in_minutes: 30
    optional: true
    device: intel_gpu
-    agent_tags:
-      label: production
-      gpu: 2+
-      mem: 24+
    no_plugin: true
    env:
      REGISTRY: "public.ecr.aws/q9t5s3a7"
@@ -42,10 +38,6 @@ steps:
    timeout_in_minutes: 30
    optional: true
    device: intel_gpu
-    agent_tags:
-      label: production
-      gpu: 1+
-      mem: 16+
    no_plugin: true
    env:
      REGISTRY: "public.ecr.aws/q9t5s3a7"
@@ -63,10 +55,6 @@ steps:
    timeout_in_minutes: 30
    optional: true
    device: intel_gpu
-    agent_tags:
-      label: production
-      gpu: 1+
-      mem: 16+
    no_plugin: true
    env:
      REGISTRY: "public.ecr.aws/q9t5s3a7"
@@ -5,10 +5,6 @@ steps:
 - label: XPU Sleep Mode
  timeout_in_minutes: 30
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -5,10 +5,6 @@ steps:
 - label: Engine (1 GPU)
  timeout_in_minutes: 30
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -6,10 +6,6 @@ steps:
  key: eplb-algorithm
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -5,10 +5,6 @@ steps:
 - label: vLLM IR Tests
  timeout_in_minutes: 30
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -5,10 +5,6 @@ steps:
 - label: LoRA Runtime + Utils
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 24+
  no_plugin: true
  working_dir: "."
  env:
@@ -38,10 +34,6 @@ steps:
 - label: LoRA Fused/MoE Kernels
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -62,10 +54,6 @@ steps:
 - label: LoRA Punica Kernels
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -86,10 +74,6 @@ steps:
 - label: LoRA Punica FP8/XPU Ops
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -110,10 +94,6 @@ steps:
 - label: LoRA Models
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 2+
-    mem: 24+
  no_plugin: true
  working_dir: "."
  env:
@@ -137,10 +117,6 @@ steps:
 - label: LoRA Multimodal
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -5,10 +5,6 @@ steps:
 - label: V1 Core + KV + Metrics
  timeout_in_minutes: 30
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -35,10 +31,6 @@ steps:
 - label: V1 Sample + Logits
  timeout_in_minutes: 30
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -79,10 +71,6 @@ steps:
 - label: XPU CPU Offload
  timeout_in_minutes: 60
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -107,10 +95,6 @@ steps:
  key: regression
  timeout_in_minutes: 30
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -142,10 +126,6 @@ steps:
  timeout_in_minutes: 30
  num_devices: 2
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 2+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -177,10 +157,6 @@ steps:
  key: async-engine-inputs-utils-worker
  timeout_in_minutes: 30
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 24+
  no_plugin: true
  working_dir: "."
  env:
@@ -5,10 +5,6 @@ steps:
 - label: Model Runner V2 Core Tests (Intel)
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 2+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -34,10 +30,6 @@ steps:
 - label: Model Runner V2 Examples (Intel)
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 24+
  no_plugin: true
  working_dir: "."
  env:
@@ -6,10 +6,6 @@ steps:
  key: multi-modal-models-standard-1-qwen2
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -31,10 +27,6 @@ steps:
  key: multi-modal-models-standard-2-qwen3-gemma
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -55,10 +47,6 @@ steps:
  key: multi-modal-models-standard-3-llava-qwen2-vl
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 24+
  no_plugin: true
  working_dir: "."
  env:
@@ -80,10 +68,6 @@ steps:
  key: multi-modal-models-standard-4-other-whisper
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -104,10 +88,6 @@ steps:
  key: multi-modal-processor
  timeout_in_minutes: 45
  device: intel_gpu
-  agent_tags:
-    label: production
-    gpu: 1+
-    mem: 16+
  no_plugin: true
  working_dir: "."
  env:
@@ -19,10 +19,6 @@ steps:
      - image-build-xpu
    timeout_in_minutes: 30
    device: intel_gpu
-    agent_tags:
-      label: production
-      gpu: 2+
-      mem: 24+
    no_plugin: true
    env:
      REGISTRY: "public.ecr.aws/q9t5s3a7"
@@ -53,10 +49,6 @@ steps:
      - image-build-xpu
    timeout_in_minutes: 30
    device: intel_gpu
-    agent_tags:
-      label: production
-      gpu: 1+
-      mem: 16+
    no_plugin: true
    env:
      REGISTRY: "public.ecr.aws/q9t5s3a7"
@@ -75,17 +67,13 @@ steps:
        pytest -v -s v1/worker --ignore=v1/worker/test_gpu_model_runner.py --ignore=v1/worker/test_worker_memory_snapshot.py &&
        pytest -v -s v1/structured_output &&
        pytest -v -s v1/test_serial_utils.py &&
-        pytest -v -s v1/spec_decode --ignore=v1/spec_decode/test_max_len.py --ignore=v1/spec_decode/test_speculators_eagle3.py --ignore=v1/spec_decode/test_acceptance_length.py --ignore=v1/spec_decode/test_speculators_correctness.py &&
+        pytest -v -s v1/spec_decode --ignore=v1/spec_decode/test_max_len.py --ignore=v1/spec_decode/test_speculators_eagle3.py --ignore=v1/spec_decode/test_acceptance_length.py &&
        pytest -v -s v1/kv_connector/unit --ignore=v1/kv_connector/unit/test_multi_connector.py --ignore=v1/kv_connector/unit/test_example_connector.py --ignore=v1/kv_connector/unit/test_lmcache_integration.py --ignore=v1/kv_connector/unit/test_hf3fs_client.py --ignore=v1/kv_connector/unit/test_hf3fs_connector.py --ignore=v1/kv_connector/unit/test_hf3fs_metadata_server.py --ignore=v1/kv_connector/unit/test_offloading_connector.py'
  - label: "XPU server test"
    depends_on:
      - image-build-xpu
    timeout_in_minutes: 30
    device: intel_gpu
-    agent_tags:
-      label: production
-      gpu: 1+
-      mem: 16+
    no_plugin: true
    env:
      REGISTRY: "public.ecr.aws/q9t5s3a7"
@@ -105,10 +93,6 @@ steps:
      - image-build-xpu
    timeout_in_minutes: 30
    device: intel_gpu
-    agent_tags:
-      label: production
-      gpu: 1+
-      mem: 16+
    no_plugin: true
    env:
      REGISTRY: "public.ecr.aws/q9t5s3a7"
@@ -4,11 +4,6 @@

 set -euo pipefail

-if python3 -c "import torch; raise SystemExit(0 if torch.version.hip is not None else 1)"; then
-    uv pip install --system -r /vllm-workspace/requirements/kv_connectors_rocm.txt
-    exit 0
-fi
-
 REQUIREMENTS_FILE="${KV_CONNECTORS_REQUIREMENTS:-/vllm-workspace/requirements/kv_connectors.txt}"

 uv pip install --system -r "${REQUIREMENTS_FILE}"
@@ -647,7 +647,7 @@ steps:
  - pytest -v -s v1/cudagraph/test_cudagraph_mode.py

 - label: e2e Core (1 GPU) # TBD
-  timeout_in_minutes: 35
+  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx90anightly, amdmi250]
  agent_pool: mi250_1
  optional: true
@@ -1594,10 +1594,9 @@ steps:
 #----------------------------------------------------------  mi300 · kernels  ----------------------------------------------------------#

 - label: Kernels Attention Test %N # TBD
-  timeout_in_minutes: 55
+  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
  agent_pool: mi300_1
-  optional: true
  parallelism: 2
  working_dir: "/vllm-workspace/tests"
  source_file_dependencies:
@@ -1628,11 +1627,10 @@ steps:
  - pytest -v -s kernels/core --ignore=kernels/core/test_minimax_reduce_rms.py  kernels/test_concat_mla_q.py kernels/test_top_k_per_row.py

 - label: Kernels MoE Test %N # TBD
-  timeout_in_minutes: 50
+  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
  agent_pool: mi300_1
-  optional: true
-  parallelism: 5
+  parallelism: 4
  working_dir: "/vllm-workspace/tests"
  source_file_dependencies:
  - csrc/quantization/cutlass_w8a8/moe/
@@ -2077,6 +2075,19 @@ steps:
  - export VLLM_ALLOW_INSECURE_SERIALIZATION=1
  - pytest -v -s v1/spec_decode/test_acceptance_length.py -m slow_test

+- label: e2e Core (1 GPU) # TBD
+  timeout_in_minutes: 180
+  mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
+  agent_pool: mi300_1
+  optional: true
+  working_dir: "/vllm-workspace/tests"
+  source_file_dependencies:
+  - vllm/v1/
+  - tests/v1/e2e/
+  - vllm/platforms/rocm.py
+  commands:
+  - pytest -v -s v1/e2e/general --ignore v1/e2e/general/test_async_scheduling.py
+
 - label: e2e Scheduling (1 GPU) # TBD
  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
@@ -2122,10 +2133,9 @@ steps:
  - pytest -v -s v1/e2e/spec_decode -k "draft_model or no_sync or batch_inference"

 - label: Spec Decode Eagle # TBD
-  timeout_in_minutes: 45
+  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx942nightly, amdmi300]
  agent_pool: mi300_1
-  optional: true
  working_dir: "/vllm-workspace/tests"
  source_file_dependencies:
  - vllm/v1/spec_decode/
@@ -3043,7 +3053,7 @@ steps:
 #----------------------------------------------------------  mi355 · kernels  ----------------------------------------------------------#

 - label: Kernels (B200-MI355) # TBD
-  timeout_in_minutes: 15
+  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx950nightly, amdmi355]
  agent_pool: mi355_1
  working_dir: "/vllm-workspace/"
@@ -3067,10 +3077,11 @@ steps:
  - pytest -v -s tests/kernels/attention/test_attention_selector.py

 - label: Kernels Attention Test %N # TBD
-  timeout_in_minutes: 60
+  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx950nightly, amdmi355]
  agent_pool: mi355_1
  parallelism: 2
+  optional: true
  working_dir: "/vllm-workspace/tests"
  source_file_dependencies:
  - csrc/attention/
@@ -3084,10 +3095,10 @@ steps:
  - pytest -v -s kernels/attention --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT

 - label: Kernels MoE Test %N # TBD
-  timeout_in_minutes: 50
+  timeout_in_minutes: 180
  mirror_hardwares: [amdexperimental, amdproduction, amdgfx950nightly, amdmi355]
  agent_pool: mi355_1
-  parallelism: 5
+  parallelism: 4
  working_dir: "/vllm-workspace/tests"
  source_file_dependencies:
  - csrc/quantization/cutlass_w8a8/moe/
@@ -74,16 +74,6 @@ steps:
    - tests/v1/e2e/general/
  commands:
    - pytest -v -s v1/e2e/general --ignore v1/e2e/general/test_async_scheduling.py
-  mirror:
-    amd:
-      device: mi250_1
-      timeout_in_minutes: 35
-      depends_on:
-      - image-build-amd
-      source_file_dependencies:
-      - vllm/v1/
-      - tests/v1/e2e/general/
-      - vllm/platforms/rocm.py

 - label: V1 e2e (2 GPUs)
  key: v1-e2e-2-gpus
@@ -74,20 +74,6 @@ steps:
  commands:
    - pytest -v -s kernels/attention --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
  parallelism: 2
-  mirror:
-    amd:
-      device: mi325_1
-      timeout_in_minutes: 55
-      depends_on:
-      - image-build-amd
-      source_file_dependencies:
-      - csrc/attention/
-      - vllm/v1/attention
-      - vllm/model_executor/layers/attention
-      - tests/kernels/attention
-      - vllm/_aiter_ops.py
-      - vllm/envs.py
-      - vllm/platforms/rocm.py

 - label: Kernels Attention DiffKV Test (H100)
  key: kernels-attention-diffkv-test-h100
@@ -118,7 +104,6 @@ steps:
      source_file_dependencies:
      - csrc/quantization/
      - vllm/model_executor/layers/quantization
-      - vllm/config/
      - tests/kernels/quantization
      - tests/kernels/quantization/test_rocm_skinny_gemms.py
      - vllm/_aiter_ops.py
@@ -142,22 +127,6 @@ steps:
    - pytest -v -s kernels/moe --ignore=kernels/moe/test_modular_oai_triton_moe.py --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
    - pytest -v -s kernels/moe/test_modular_oai_triton_moe.py --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
  parallelism: 5
-  mirror:
-    amd:
-      device: mi325_1
-      timeout_in_minutes: 50
-      source_file_dependencies:
-      - csrc/quantization/cutlass_w8a8/moe/
-      - csrc/moe/
-      - tests/kernels/moe
-      - vllm/model_executor/layers/fused_moe/
-      - vllm/distributed/device_communicators/
-      - vllm/envs.py
-      - vllm/config
-      - vllm/_aiter_ops.py
-      - vllm/platforms/rocm.py
-      depends_on:
-      - image-build-amd

 - label: Kernels Mamba Test
  key: kernels-mamba-test
@@ -101,16 +101,6 @@ steps:
  num_devices: 8
  commands:
    - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-h200.txt
-  mirror:
-    amd:
-      device: mi300_8
-      timeout_in_minutes: 180
-      depends_on:
-      - image-build-amd
-      commands:
-      - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-      - export PYTORCH_ROCM_ARCH=gfx942 # Limit Quark compilation to save time
-      - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-mi3xx.txt

 - label: MoE Refactor Integration Test (H100 - TEMPORARY)
  key: moe-refactor-integration-test-h100-temporary
@@ -105,12 +105,6 @@ steps:
    # Integration test for streaming correctness (requires special branch).
    - pip install -U git+https://github.com/robertgshaw2-redhat/lm-evaluation-harness.git@streaming-api
    - pytest -v -s entrypoints/openai/correctness/test_lmeval.py::test_lm_eval_accuracy_v1_engine
-  mirror:
-    amd:
-      device: mi325_1
-      timeout_in_minutes: 60
-      depends_on:
-      - image-build-amd

 - label: V1 Others (CPU)
  key: v1-others-cpu
@@ -68,6 +68,7 @@ steps:
    - cd .. && VLLM_WORKER_MULTIPROC_METHOD=spawn pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model  # Otherwise, mp_method="spawn" doesn't work
  mirror:
    amd:
+      soft_fail: true
      device: mi325_1
      depends_on:
      - image-build-amd
@@ -107,12 +107,6 @@ steps:
    - tests/compile/passes
  commands:
  - pytest -s -v compile/passes --ignore compile/passes/distributed
-  mirror:
-    amd:
-      device: mi300_1
-      timeout_in_minutes: 180
-      depends_on:
-      - image-build-amd

 - label: PyTorch Fullgraph Smoke Test
  key: pytorch-fullgraph-smoke-test
@@ -12,20 +12,6 @@ steps:
    - tests/v1/e2e/spec_decode/
  commands:
    - pytest -v -s v1/e2e/spec_decode -k "eagle_correctness"
-  mirror:
-    amd:
-      device: mi325_1
-      timeout_in_minutes: 45
-      depends_on:
-      - image-build-amd
-      source_file_dependencies:
-      - vllm/v1/spec_decode/
-      - vllm/v1/worker/gpu/spec_decode/
-      - vllm/model_executor/model_loader/
-      - vllm/v1/sample/
-      - vllm/model_executor/layers/
-      - tests/v1/e2e/spec_decode/
-      - vllm/platforms/rocm.py

 - label: Spec Decode Eagle Nightly B200
  key: spec-decode-eagle-nightly-b200
@@ -1,35 +0,0 @@
---
-name: ci-fails-buildkite
-description: Fetch and diagnose vLLM Buildkite CI failure logs. Use when investigating failing CI jobs on a PR or build, when the user pastes a buildkite.com URL, or asks to fetch/diagnose CI logs.
---
-
-# Diagnosing vLLM Buildkite CI Failures
-
-Buildkite logs are public; no login needed.
-
-`.buildkite/scripts/ci-fetch-log.sh` saves each log as `ci-<build>-<job-name>.log`, stripped of timestamps and ANSI codes. Existing files are kept; set `CI_FETCH_LOG_FORCE=1` to refetch.
-
-## Fetching logs
-
-```bash
-# All failed jobs in a PR's latest build (current branch's PR if omitted):
-.buildkite/scripts/ci-fetch-log.sh --pr <PR>
-
-# All failed jobs in a build (--soft also includes soft-failed jobs;
-# --all fetches every finished job):
-.buildkite/scripts/ci-fetch-log.sh "https://buildkite.com/vllm/ci/builds/<N>"
-
-# One job — `gh pr checks` URLs (#<job_uuid>) and web UI URLs (?sid=) both
-# work; pass "-" as a second argument to stream to stdout:
-.buildkite/scripts/ci-fetch-log.sh "https://buildkite.com/vllm/ci/builds/<N>#<job_uuid>"
-```
-
-To clean an already-downloaded log with `.buildkite/scripts/ci-clean-log.sh`:
-
-```bash
-./ci-clean-log.sh ci.log
-```
-
-## Reference
-
-See [docs/contributing/ci/failures.md](../../../docs/contributing/ci/failures.md) for the full guide: filing CI failure issues, investigating/bisecting, reproducing flaky tests, and daily triage.
@@ -2,14 +2,15 @@
 # for more info about CODEOWNERS file

 # This lists cover the "core" components of vLLM that require careful review
-/vllm/compilation @zou3519 @youkaichao @ProExpertProg @BoyuanFeng
+/vllm/compilation @zou3519 @youkaichao @ProExpertProg @BoyuanFeng @vadiklyutiy
 /vllm/distributed/kv_transfer @NickLucche @ApostaC @orozery @xuechendi
 /vllm/lora @jeejeelee
 /vllm/model_executor/layers/attention @LucasWilkinson @MatthewBonanni
 /vllm/model_executor/layers/fused_moe @mgoin @pavanimajety @zyongye
 /vllm/model_executor/layers/quantization @mgoin @robertgshaw2-redhat @tlrmchlsmth @yewentao256 @pavanimajety @zyongye
 /vllm/model_executor/layers/mamba @tdoublep @tomeras91
-/vllm/model_executor/layers/mamba/gdn/qwen_gdn_linear_attn.py @tdoublep @ZJY0516 @vadiklyutiy
+/vllm/model_executor/layers/mamba/gdn_linear_attn.py @tdoublep @ZJY0516 @vadiklyutiy
+/vllm/model_executor/layers/rotary_embedding.py @vadiklyutiy
 /vllm/model_executor/model_loader @22quinn
 /vllm/model_executor/layers/batch_invariant.py @yewentao256 
 /vllm/ir @ProExpertProg
@@ -199,9 +199,7 @@ cython_debug/
 .vscode/

 # Claude
-.claude/*
-!.claude/skills/
-!.claude/skills/**
+.claude/

 # Codex
 .codex/
@@ -114,6 +114,17 @@ Follow these rules for all code changes in this repository:
 - Keep comments and docstrings minimal and concise.
 - Assume the reader is familiar with vLLM.

+### Diagnosing CI failures
+
+Buildkite logs are public; no login needed. Details: [docs/contributing/ci/failures.md](docs/contributing/ci/failures.md).
+
+```bash
+# All failed-job logs for a PR's latest build (current branch's PR if omitted):
+.buildkite/scripts/ci-fetch-log.sh --pr <PR>
+# Any Buildkite build or job URL also works:
+.buildkite/scripts/ci-fetch-log.sh "<buildkite_url>"
+```
+
 ### Commit messages

 Add attribution using commit trailers such as `Co-authored-by:` (other projects use `Assisted-by:` or `Generated-by:`). For example:
@@ -319,35 +319,82 @@ if(VLLM_GPU_LANG STREQUAL "CUDA" OR VLLM_GPU_LANG STREQUAL "HIP")
 endif()

 #
-# Legacy _C extension (ROCm only — CUDA ops migrated to _C_stable_libtorch)
+# _C extension
 #

-if(VLLM_GPU_LANG STREQUAL "HIP")
-  set(VLLM_EXT_SRC
-    "csrc/torch_bindings.cpp"
+set(VLLM_EXT_SRC
+  "csrc/quantization/activation_kernels.cu"
+  "csrc/torch_bindings.cpp")
+
+if(VLLM_GPU_LANG STREQUAL "CUDA")
+  SET(CUTLASS_ENABLE_HEADERS_ONLY ON CACHE BOOL "Enable only the header library")
+
+  # Set CUTLASS_REVISION. Used for FetchContent. Also fixes some bogus messages when building.
+  set(CUTLASS_REVISION "v4.4.2")
+
+  # Use the specified CUTLASS source directory for compilation if VLLM_CUTLASS_SRC_DIR is provided
+  if (DEFINED ENV{VLLM_CUTLASS_SRC_DIR})
+    set(VLLM_CUTLASS_SRC_DIR $ENV{VLLM_CUTLASS_SRC_DIR})
+  endif()
+
+  if(VLLM_CUTLASS_SRC_DIR)
+    if(NOT IS_ABSOLUTE VLLM_CUTLASS_SRC_DIR)
+      get_filename_component(VLLM_CUTLASS_SRC_DIR "${VLLM_CUTLASS_SRC_DIR}" ABSOLUTE)
+    endif()
+    message(STATUS "The VLLM_CUTLASS_SRC_DIR is set, using ${VLLM_CUTLASS_SRC_DIR} for compilation")
+    FetchContent_Declare(cutlass SOURCE_DIR ${VLLM_CUTLASS_SRC_DIR})
+  else()
+    FetchContent_Declare(
+        cutlass
+        GIT_REPOSITORY https://github.com/nvidia/cutlass.git
+        # Please keep this in sync with CUTLASS_REVISION line above.
+        GIT_TAG ${CUTLASS_REVISION}
+        GIT_PROGRESS TRUE
+
+        # Speed up CUTLASS download by retrieving only the specified GIT_TAG instead of the history.
+        # Important: If GIT_SHALLOW is enabled then GIT_TAG works only with branch names and tags.
+        # So if the GIT_TAG above is updated to a commit hash, GIT_SHALLOW must be set to FALSE
+        GIT_SHALLOW TRUE
+    )
+  endif()
+  FetchContent_MakeAvailable(cutlass)
+
+  set_gencode_flags_for_srcs(
+    SRCS "${VLLM_EXT_SRC}"
+    CUDA_ARCHS "${CUDA_ARCHS}")
+
+# if CUDA endif
+endif()
+
+if (VLLM_GPU_LANG STREQUAL "HIP")
+  # Add QuickReduce kernels (ROCm-only; not part of stable ABI migration).
+  # TODO: Remove the cuda_view when ROCm upgrade to torch 2.11.
+  list(APPEND VLLM_EXT_SRC
    "csrc/custom_quickreduce.cu"
    "csrc/cuda_view.cu"
-    "csrc/libtorch_stable/cuda_utils_kernels.cu")
+    "csrc/libtorch_stable/cuda_utils_kernels.cu"
+  )
+# if ROCM endif
+endif()

-  message(STATUS "Enabling C extension.")
-  define_extension_target(
-    _C
-    DESTINATION vllm
-    LANGUAGE ${VLLM_GPU_LANG}
-    SOURCES ${VLLM_EXT_SRC}
-    COMPILE_FLAGS ${VLLM_GPU_FLAGS}
-    ARCHITECTURES ${VLLM_GPU_ARCHES}
-    INCLUDE_DIRECTORIES ${CUTLASS_INCLUDE_DIR}
-    INCLUDE_DIRECTORIES ${CUTLASS_TOOLS_UTIL_INCLUDE_DIR}
-    USE_SABI 3
-    WITH_SOABI)
+message(STATUS "Enabling C extension.")
+define_extension_target(
+  _C
+  DESTINATION vllm
+  LANGUAGE ${VLLM_GPU_LANG}
+  SOURCES ${VLLM_EXT_SRC}
+  COMPILE_FLAGS ${VLLM_GPU_FLAGS}
+  ARCHITECTURES ${VLLM_GPU_ARCHES}
+  INCLUDE_DIRECTORIES ${CUTLASS_INCLUDE_DIR}
+  INCLUDE_DIRECTORIES ${CUTLASS_TOOLS_UTIL_INCLUDE_DIR}
+  USE_SABI 3
+  WITH_SOABI)

-  # If CUTLASS is compiled on NVCC >= 12.5, it by default uses
-  # cudaGetDriverEntryPointByVersion as a wrapper to avoid directly calling the
-  # driver API. This causes problems when linking with earlier versions of CUDA.
-  # Setting this variable sidesteps the issue by calling the driver directly.
-  target_compile_definitions(_C PRIVATE CUTLASS_ENABLE_DIRECT_CUDA_DRIVER_CALL=1)
-endif() # _C HIP endif
+# If CUTLASS is compiled on NVCC >= 12.5, it by default uses
+# cudaGetDriverEntryPointByVersion as a wrapper to avoid directly calling the
+# driver API. This causes problems when linking with earlier versions of CUDA.
+# Setting this variable sidesteps the issue by calling the driver directly.
+target_compile_definitions(_C PRIVATE CUTLASS_ENABLE_DIRECT_CUDA_DRIVER_CALL=1)

 if(VLLM_GPU_LANG STREQUAL "CUDA" OR VLLM_GPU_LANG STREQUAL "HIP")
  #
@@ -356,7 +403,6 @@ if(VLLM_GPU_LANG STREQUAL "CUDA" OR VLLM_GPU_LANG STREQUAL "HIP")
  set(VLLM_STABLE_EXT_SRC
    "csrc/libtorch_stable/torch_bindings.cpp"
    "csrc/libtorch_stable/activation_kernels.cu"
-    "csrc/libtorch_stable/quantization/activation_kernels.cu"
    "csrc/libtorch_stable/quantization/w8a8/int8/scaled_quant.cu"
    "csrc/libtorch_stable/quantization/w8a8/fp8/common.cu"
    "csrc/libtorch_stable/quantization/w8a8/fp8/per_token_group_quant.cu"
@@ -383,38 +429,6 @@ if(VLLM_GPU_LANG STREQUAL "CUDA" OR VLLM_GPU_LANG STREQUAL "HIP")
    "csrc/libtorch_stable/fused_deepseek_v4_qnorm_rope_kv_insert_kernel.cu")

  if(VLLM_GPU_LANG STREQUAL "CUDA")
-    SET(CUTLASS_ENABLE_HEADERS_ONLY ON CACHE BOOL "Enable only the header library")
-
-    # Set CUTLASS_REVISION. Used for FetchContent. Also fixes some bogus messages when building.
-    set(CUTLASS_REVISION "v4.4.2")
-
-    # Use the specified CUTLASS source directory for compilation if VLLM_CUTLASS_SRC_DIR is provided
-    if (DEFINED ENV{VLLM_CUTLASS_SRC_DIR})
-      set(VLLM_CUTLASS_SRC_DIR $ENV{VLLM_CUTLASS_SRC_DIR})
-    endif()
-
-    if(VLLM_CUTLASS_SRC_DIR)
-      if(NOT IS_ABSOLUTE VLLM_CUTLASS_SRC_DIR)
-        get_filename_component(VLLM_CUTLASS_SRC_DIR "${VLLM_CUTLASS_SRC_DIR}" ABSOLUTE)
-      endif()
-      message(STATUS "The VLLM_CUTLASS_SRC_DIR is set, using ${VLLM_CUTLASS_SRC_DIR} for compilation")
-      FetchContent_Declare(cutlass SOURCE_DIR ${VLLM_CUTLASS_SRC_DIR})
-    else()
-      FetchContent_Declare(
-          cutlass
-          GIT_REPOSITORY https://github.com/nvidia/cutlass.git
-          # Please keep this in sync with CUTLASS_REVISION line above.
-          GIT_TAG ${CUTLASS_REVISION}
-          GIT_PROGRESS TRUE
-
-          # Speed up CUTLASS download by retrieving only the specified GIT_TAG instead of the history.
-          # Important: If GIT_SHALLOW is enabled then GIT_TAG works only with branch names and tags.
-          # So if the GIT_TAG above is updated to a commit hash, GIT_SHALLOW must be set to FALSE
-          GIT_SHALLOW TRUE
-      )
-    endif()
-    FetchContent_MakeAvailable(cutlass)
-
    list(APPEND VLLM_STABLE_EXT_SRC
      "csrc/libtorch_stable/cuda_view.cu"
      "csrc/libtorch_stable/cuda_utils_kernels.cu"
@@ -915,6 +929,7 @@ if(VLLM_GPU_LANG STREQUAL "CUDA" OR VLLM_GPU_LANG STREQUAL "HIP")
      SRCS "${FP4_SM120_SRCS}"
      CUDA_ARCHS "${FP4_SM120_ARCHS}")
    list(APPEND VLLM_STABLE_EXT_SRC "${FP4_SM120_SRCS}")
+    target_compile_definitions(_C PRIVATE ENABLE_NVFP4_SM120=1)
    list(APPEND VLLM_GPU_FLAGS "-DENABLE_NVFP4_SM120=1")
    list(APPEND VLLM_GPU_FLAGS "-DENABLE_CUTLASS_MOE_SM120=1")
    message(STATUS "Building SM12x NVFP4 for archs: ${FP4_SM120_ARCHS}")
@@ -947,6 +962,7 @@ if(VLLM_GPU_LANG STREQUAL "CUDA" OR VLLM_GPU_LANG STREQUAL "HIP")
      SRCS "${FP4_SM100_SRCS}"
      CUDA_ARCHS "${FP4_SM100_ARCHS}")
    list(APPEND VLLM_STABLE_EXT_SRC "${FP4_SM100_SRCS}")
+    target_compile_definitions(_C PRIVATE ENABLE_NVFP4_SM100=1)
    list(APPEND VLLM_GPU_FLAGS "-DENABLE_NVFP4_SM100=1")
    list(APPEND VLLM_GPU_FLAGS "-DENABLE_CUTLASS_MOE_SM100=1")
    message(STATUS "Building SM10x/11x NVFP4/MXFP4 for archs: ${FP4_SM100_ARCHS}")
@@ -60,7 +60,6 @@ endif()

 if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER_EQUAL 12.8 AND QUTLASS_ARCHS)
  set(QUTLASS_SOURCES
-    csrc/qutlass_registration.cpp
    ${qutlass_SOURCE_DIR}/qutlass/csrc/bindings.cpp
    ${qutlass_SOURCE_DIR}/qutlass/csrc/gemm.cu
    ${qutlass_SOURCE_DIR}/qutlass/csrc/gemm_ada.cu
@@ -79,19 +78,8 @@ if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER_EQUAL 12.8 AND QUTLASS_ARCHS)

  if(CUTLASS_INCLUDE_DIR AND EXISTS "${CUTLASS_INCLUDE_DIR}/cutlass/cutlass.h")
    list(APPEND QUTLASS_INCLUDES "${CUTLASS_INCLUDE_DIR}")
-    if(CUTLASS_TOOLS_UTIL_INCLUDE_DIR AND
-       EXISTS "${CUTLASS_TOOLS_UTIL_INCLUDE_DIR}/cutlass/util/packed_stride.hpp")
-      list(APPEND QUTLASS_INCLUDES "${CUTLASS_TOOLS_UTIL_INCLUDE_DIR}")
-    else()
-      get_filename_component(_qutlass_cutlass_root "${CUTLASS_INCLUDE_DIR}" DIRECTORY)
-      if(EXISTS "${_qutlass_cutlass_root}/tools/util/include/cutlass/util/packed_stride.hpp")
-        list(APPEND QUTLASS_INCLUDES "${_qutlass_cutlass_root}/tools/util/include")
-      endif()
-    endif()
  elseif(EXISTS "${qutlass_SOURCE_DIR}/qutlass/third_party/cutlass/include/cutlass/cutlass.h")
-    list(APPEND QUTLASS_INCLUDES
-      "${qutlass_SOURCE_DIR}/qutlass/third_party/cutlass/include"
-      "${qutlass_SOURCE_DIR}/qutlass/third_party/cutlass/tools/util/include")
+    list(APPEND QUTLASS_INCLUDES "${qutlass_SOURCE_DIR}/qutlass/third_party/cutlass/include")
    message(STATUS "[QUTLASS] Using QuTLASS vendored CUTLASS headers (no vLLM CUTLASS detected).")
  else()
    message(FATAL_ERROR "[QUTLASS] CUTLASS headers not found. "
@@ -103,23 +91,12 @@ if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER_EQUAL 12.8 AND QUTLASS_ARCHS)
    CUDA_ARCHS "${QUTLASS_ARCHS}"
  )

-  # QuTLASS uses legacy ATen headers and cannot be built with TORCH_TARGET_VERSION.
-  # Keep it as its own extension (registers torch.ops._qutlass_C).
-  define_extension_target(
-    _qutlass_C
-    DESTINATION vllm
-    LANGUAGE ${VLLM_GPU_LANG}
-    SOURCES ${QUTLASS_SOURCES}
-    COMPILE_FLAGS ${VLLM_GPU_FLAGS}
-    ARCHITECTURES ${VLLM_GPU_ARCHES}
-    INCLUDE_DIRECTORIES ${QUTLASS_INCLUDES}
-    USE_SABI 3
-    WITH_SOABI)
-
-  target_compile_definitions(_qutlass_C PRIVATE
+  target_sources(_C PRIVATE ${QUTLASS_SOURCES})
+  target_include_directories(_C PRIVATE ${QUTLASS_INCLUDES})
+  target_compile_definitions(_C PRIVATE
    QUTLASS_DISABLE_PYBIND=1
    TARGET_CUDA_ARCH=${QUTLASS_TARGET_CC}
-    CUTLASS_ENABLE_DIRECT_CUDA_DRIVER_CALL=1)
+  )

  set_property(SOURCE ${QUTLASS_SOURCES} APPEND PROPERTY COMPILE_OPTIONS
    $<$<COMPILE_LANGUAGE:CUDA>:--expt-relaxed-constexpr --use_fast_math -O3>
@@ -134,5 +111,4 @@ else()
      "[QUTLASS] Skipping build: no supported arch (12.0f / 10.0f) found in "
      "CUDA_ARCHS='${CUDA_ARCHS}'.")
  endif()
-  add_custom_target(_qutlass_C)
 endif()
@@ -268,14 +268,9 @@ int64_t sm100_cutlass_mla_get_workspace_size(int64_t max_seq_len, int64_t num_ba
  using TileShapeD = typename MlaSm100Type::TileShapeD;
  arguments.problem_shape =
      cute::make_tuple(TileShapeH{}, static_cast<int>(max_seq_len), TileShapeD{}, static_cast<int>(num_batches));
-  if (sm_count <= 0) {
-    int current_device = 0;
-    cudaGetDevice(&current_device);
-    arguments.hw_info.sm_count =
-        cutlass::KernelHardwareInfo::query_device_multiprocessor_count(current_device);
-  } else {
-    arguments.hw_info.sm_count = sm_count;
-  }
+  // Assumes device 0 when getting sm_count.
+  arguments.hw_info.sm_count =
+      sm_count <= 0 ? cutlass::KernelHardwareInfo::query_device_multiprocessor_count(/*device_id=*/0) : sm_count;
  arguments.split_kv = static_cast<int>(num_kv_splits);
  MlaSm100Type::Fmha::set_split_kv(arguments);

@@ -81,11 +81,11 @@ __global__ void rms_norm_kernel(
 #pragma unroll
    for (int j = 0; j < VEC_SIZE; j++) {
      float x = static_cast<float>(src1.val[j]);
+      scalar_t normalized = static_cast<scalar_t>(x * s_variance);
      if constexpr (HasWeight) {
-        float w = static_cast<float>(src2.val[j]);
-        dst.val[j] = static_cast<scalar_t>(x * s_variance * w);
+        dst.val[j] = normalized * src2.val[j];
      } else {
-        dst.val[j] = static_cast<scalar_t>(x * s_variance);
+        dst.val[j] = normalized;
      }
    }
    v_out[i] = dst;
@@ -151,8 +151,7 @@ fused_add_rms_norm_kernel(
 #pragma unroll
      for (int j = 0; j < width; ++j) {
        float x = Converter::convert(res.data[j]);
-        float wf = Converter::convert(w.data[j]);
-        out.data[j] = Converter::convert(x * s_variance * wf);
+        out.data[j] = Converter::convert(x * s_variance) * w.data[j];
      }
    } else {
 #pragma unroll
@@ -199,8 +198,8 @@ fused_add_rms_norm_kernel(
  for (int idx = threadIdx.x; idx < hidden_size; idx += blockDim.x) {
    float x = (float)residual[blockIdx.x * hidden_size + idx];
    if constexpr (HasWeight) {
-      float w = (float)weight[idx];
-      input[blockIdx.x * input_stride + idx] = (scalar_t)(x * s_variance * w);
+      input[blockIdx.x * input_stride + idx] =
+          (scalar_t)(x * s_variance) * weight[idx];
    } else {
      input[blockIdx.x * input_stride + idx] = (scalar_t)(x * s_variance);
    }
@@ -66,13 +66,8 @@ __global__ void rms_norm_static_fp8_quant_kernel(
 #pragma unroll
    for (int j = 0; j < VEC_SIZE; j++) {
      float x = static_cast<float>(src1.val[j]);
-      float w = static_cast<float>(src2.val[j]);
-      // Round normalized result through scalar_t to match the precision of the
-      // unfused composite (rms_norm writes scalar_t, then
-      // static_scaled_fp8_quant re-loads it as float before FP8 conversion).
-      // Without this round, the fused path is strictly more accurate and
-      // disagrees with the composite at exact E4M3 quantization tie boundaries.
-      scalar_t out_norm = static_cast<scalar_t>(x * s_variance * w);
+      // Multiply in weight's native dtype to match rms_norm_kernel.
+      scalar_t out_norm = static_cast<scalar_t>(x * s_variance) * src2.val[j];
      out[blockIdx.x * hidden_size + idx * VEC_SIZE + j] =
          scaled_fp8_conversion<true, fp8_type>(static_cast<float>(out_norm),
                                                scale_inv);
@@ -142,12 +137,8 @@ fused_add_rms_norm_static_fp8_quant_kernel(
 #pragma unroll
    for (int i = 0; i < width; ++i) {
      float x = Converter::convert(res.data[i]);
-      float wf = Converter::convert(w.data[i]);
-      // See note in rms_norm_static_fp8_quant_kernel: round through scalar_t
-      // to match the unfused composite path at FP8 boundaries. We use the
-      // backend's hip_type for the intermediate since c10::Half/BFloat16 has
-      // ambiguous conversions on CUDA and no implicit conversion on ROCm.
-      HipT out_norm_h = Converter::convert(x * s_variance * wf);
+      // Multiply in weight's native dtype to match fused_add_rms_norm_kernel.
+      HipT out_norm_h = Converter::convert(x * s_variance) * w.data[i];
      out[id * width + i] = scaled_fp8_conversion<true, fp8_type>(
          Converter::convert(out_norm_h), scale_inv);
    }
@@ -192,10 +183,8 @@ fused_add_rms_norm_static_fp8_quant_kernel(

  for (int idx = threadIdx.x; idx < hidden_size; idx += blockDim.x) {
    float x = (float)residual[blockIdx.x * hidden_size + idx];
-    float w = (float)weight[idx];
-    // See note in rms_norm_static_fp8_quant_kernel: round through scalar_t
-    // to match the unfused composite path at FP8 boundaries.
-    scalar_t out_norm = static_cast<scalar_t>(x * s_variance * w);
+    // Multiply in weight's native dtype to match fused_add_rms_norm_kernel.
+    scalar_t out_norm = static_cast<scalar_t>(x * s_variance) * weight[idx];
    out[blockIdx.x * hidden_size + idx] = scaled_fp8_conversion<true, fp8_type>(
        static_cast<float>(out_norm), scale_inv);
  }
@@ -9,7 +9,7 @@
 #include <torch/headeronly/core/ScalarType.h>

 #include "../../cuda_compat.h"
-#include "libtorch_stable/core/math.hpp"
+#include "core/math.hpp"
 #include "libtorch_stable/dispatch_utils.h"
 #include "libtorch_stable/torch_utils.h"

@@ -2,25 +2,9 @@

 #include <torch/csrc/stable/library.h>
 #include <torch/csrc/stable/tensor.h>
-#include <torch/headeronly/util/Exception.h>

 #include <optional>
 #include <string>
-#include <vector>
-
-#include <torch/csrc/stable/ops.h>
-
-inline torch::stable::Tensor weak_ref_tensor(torch::stable::Tensor& tensor) {
-  // Ensure tensor is on CUDA
-  STD_TORCH_CHECK(tensor.device().is_cuda(), "Tensor must be on CUDA device");
-
-  // Get the raw data pointer
-  void* data_ptr = tensor.mutable_data_ptr();
-
-  /// Create a new tensor from the raw data pointer
-  return torch::stable::from_blob(data_ptr, tensor.sizes(), tensor.strides(),
-                                  tensor.device(), tensor.scalar_type());
-}

 void per_token_group_quant_fp8(const torch::stable::Tensor& input,
                               torch::stable::Tensor& output_q,
@@ -387,18 +371,6 @@ void silu_and_mul(torch::stable::Tensor& out, torch::stable::Tensor& input);
 void silu_and_mul_clamp(torch::stable::Tensor& out,
                        torch::stable::Tensor& input, double limit,
                        double alpha = 1.0, double beta = 0.0);
-
-void silu_and_mul_quant(torch::stable::Tensor& out,
-                        torch::stable::Tensor& input,
-                        torch::stable::Tensor& scale);
-
-void persistent_masked_m_silu_mul_quant(
-    const torch::stable::Tensor& input,              // (E, T, 2*H)
-    const torch::stable::Tensor& tokens_per_expert,  // (E)
-    torch::stable::Tensor& y_q,                      // (E, T, H) [OUT]
-    torch::stable::Tensor& y_s,  // (E, T, H//group_size) [OUT]
-    bool use_ue8m0);
-
 void mul_and_silu(torch::stable::Tensor& out, torch::stable::Tensor& input);
 void gelu_and_mul(torch::stable::Tensor& out, torch::stable::Tensor& input);
 void gelu_tanh_and_mul(torch::stable::Tensor& out,
@@ -31,7 +31,7 @@

 #include "cutlass/util/packed_stride.hpp"

-#include "libtorch_stable/core/math.hpp"
+#include "core/math.hpp"
 #include "core/batch_invariant.hpp"

 using namespace cute;
@@ -31,7 +31,7 @@

 #include "cutlass/util/packed_stride.hpp"

-#include "libtorch_stable/core/math.hpp"
+#include "core/math.hpp"
 #include "core/batch_invariant.hpp"

 using namespace cute;
@@ -19,7 +19,7 @@
 #include "cutlass/gemm/collective/collective_builder.hpp"
 #include "cutlass/util/packed_stride.hpp"

-#include "libtorch_stable/core/math.hpp"
+#include "core/math.hpp"
 #include "libtorch_stable/cutlass_extensions/common.hpp"
 // clang-format on

@@ -14,7 +14,7 @@
 #include "cutlass/epilogue/collective/collective_builder.hpp"
 #include "cutlass/gemm/collective/collective_builder.hpp"

-#include "libtorch_stable/core/math.hpp"
+#include "core/math.hpp"
 #include "libtorch_stable/cutlass_extensions/common.hpp"
 // clang-format on

@@ -22,7 +22,7 @@
 #include "cutlass/epilogue/threadblock/fusion/visitors.hpp"
 #include "cutlass/gemm/kernel/default_gemm_universal_with_visitor.h"

-#include "libtorch_stable/core/math.hpp"
+#include "core/math.hpp"
 #include "libtorch_stable/cutlass_extensions/common.hpp"
 // clang-format on

@@ -301,9 +301,8 @@ __global__ void per_token_group_quant_8bit_packed_register_kernel(

  const int sf_k_local = local_group_id % kGroupsPerBlockX;
  const int row_local = local_group_id / kGroupsPerBlockX;
-  // Rows on grid.x: mn scales with tokens and can exceed the 65535 grid.y cap.
-  const int sf_k_idx = blockIdx.y * kGroupsPerBlockX + sf_k_local;
-  const int mn_idx = blockIdx.x * kRowsPerBlock + row_local;
+  const int sf_k_idx = blockIdx.x * kGroupsPerBlockX + sf_k_local;
+  const int mn_idx = blockIdx.y * kRowsPerBlock + row_local;

 #if (defined(__CUDA_ARCH__) && (__CUDA_ARCH__ >= 900))
  asm volatile("griddepcontrol.wait;");
@@ -497,15 +496,14 @@ void per_token_group_quant_8bit_packed(const torch::stable::Tensor& input,
                  " is not a multiple of 4.");
  const int kx = GetGroupsPerBlockX(padded_groups_per_row);
  const int ry = 16 / kx;
-  const int64_t row_blocks = (tma_aligned_mn + ry - 1) / ry;
-  const int64_t sf_k_blocks = padded_groups_per_row / kx;
+  const int64_t blocks_x = padded_groups_per_row / kx;
+  const int64_t blocks_y = (tma_aligned_mn + ry - 1) / ry;
  const int num_threads = (kx * ry) * THREADS_PER_GROUP;
-  // CUDA caps grid.x at 2^31 - 1 and grid.y at 2^16 - 1 (65535).
-  constexpr int64_t kMaxGridDimYZ = 65535;
-  STD_TORCH_CHECK(row_blocks <= static_cast<int64_t>(INT32_MAX) &&
-                      sf_k_blocks <= kMaxGridDimYZ,
+  // CUDA caps grid.x and grid.y at 2^31 - 1; guard against pathological inputs.
+  STD_TORCH_CHECK(blocks_x <= static_cast<int64_t>(INT32_MAX) &&
+                      blocks_y <= static_cast<int64_t>(INT32_MAX),
                  "per_token_group_quant_8bit_packed grid too large: (",
-                  row_blocks, ", ", sf_k_blocks, ").");
+                  blocks_x, ", ", blocks_y, ").");

  auto dst_type = output_q.scalar_type();

@@ -515,8 +513,8 @@ void per_token_group_quant_8bit_packed(const torch::stable::Tensor& input,
  #define LAUNCH_REG_KERNEL_INST(T, DST_DTYPE, KX, RY)                         \
    do {                                                                       \
      cudaLaunchConfig_t config = {};                                          \
-      config.gridDim = dim3(static_cast<unsigned int>(row_blocks),             \
-                            static_cast<unsigned int>(sf_k_blocks));           \
+      config.gridDim = dim3(static_cast<unsigned int>(blocks_x),               \
+                            static_cast<unsigned int>(blocks_y));              \
      config.blockDim = dim3(num_threads);                                     \
      config.dynamicSmemBytes = 0;                                             \
      config.stream = stream;                                                  \
@@ -541,8 +539,8 @@ void per_token_group_quant_8bit_packed(const torch::stable::Tensor& input,
 #else
  #define LAUNCH_REG_KERNEL_INST(T, DST_DTYPE, KX, RY)                         \
    do {                                                                       \
-      dim3 grid(static_cast<unsigned int>(row_blocks),                         \
-                static_cast<unsigned int>(sf_k_blocks));                       \
+      dim3 grid(static_cast<unsigned int>(blocks_x),                           \
+                static_cast<unsigned int>(blocks_y));                          \
      dim3 block(num_threads);                                                 \
      per_token_group_quant_8bit_packed_register_kernel<T, DST_DTYPE, 128, KX, \
                                                        RY>                    \
@@ -34,20 +34,6 @@ STABLE_TORCH_LIBRARY_FRAGMENT(_C, ops) {
  // TODO: Remove this once ROCm upgrade to torch 2.11.
  ops.def("get_cuda_view_from_cpu_tensor(Tensor cpu_tensor) -> Tensor");

-  // Note about marlin kernel 'workspace' arguments:
-  // Technically these should be mutable since they are modified by the kernel.
-  // But since they are set back to zero once the kernel is finished we can
-  // hand wave and say that they have no net effect.
-  //
-  // The reason to mark 'workspace' as immutable is so that they don't interfere
-  // with using ScalarType arguments in the ops. If they are marked as mutable,
-  // pytorch throws an assert in
-  // 'torch._higher_order_ops._register_effectful_op' that prevents these
-  // kernels from being torch.compile'd.
-  // See the following document for more info on custom types and ops that use
-  // custom types:
-  // https://docs.google.com/document/d/18fBMPuOJ0fY5ZQ6YyrHUppw9FA332CpNtgB6SOIgyuA
-
  // Machete (Dense) Optimized Mixed Precision GEMM for Hopper.
  ops.def(
      "machete_supported_schedules("
@@ -494,11 +480,6 @@ STABLE_TORCH_LIBRARY_FRAGMENT(_C, ops) {
      "Tensor workspace, int k, int max_seq_len) -> ()");

  // Activation ops
-  ops.def(
-      "persistent_masked_m_silu_mul_quant(Tensor input, Tensor counts, Tensor! "
-      "y_q, Tensor! y_s, bool use_ue8m0) -> ()");
-  ops.def("weak_ref_tensor(Tensor input) -> Tensor");
-
  // Activation function used in SwiGLU.
  ops.def("silu_and_mul(Tensor! result, Tensor input) -> ()");

@@ -511,10 +492,6 @@ STABLE_TORCH_LIBRARY_FRAGMENT(_C, ops) {
      "silu_and_mul_with_clamp(Tensor! result, Tensor input, float limit, "
      "float alpha=1.0, float beta=0.0) -> ()");

-  // SwiGLU activation with FP8 quantization.
-  ops.def(
-      "silu_and_mul_quant(Tensor! result, Tensor input, Tensor scale) -> ()");
-
  // Activation function used in GeGLU with `none` approximation.
  ops.def("gelu_and_mul(Tensor! out, Tensor input) -> ()");

@@ -713,10 +690,6 @@ STABLE_TORCH_LIBRARY_IMPL(_C, CUDA, ops) {
  ops.impl("persistent_topk", TORCH_BOX(&persistent_topk));

  // Activation kernels (shared CUDA/ROCm)
-  ops.impl("persistent_masked_m_silu_mul_quant",
-           TORCH_BOX(&persistent_masked_m_silu_mul_quant));
-  ops.impl("weak_ref_tensor", TORCH_BOX(&weak_ref_tensor));
-  ops.impl("silu_and_mul_quant", TORCH_BOX(&silu_and_mul_quant));
  ops.impl("silu_and_mul", TORCH_BOX(&silu_and_mul));
  ops.impl("mul_and_silu", TORCH_BOX(&mul_and_silu));
  ops.impl("gelu_and_mul", TORCH_BOX(&gelu_and_mul));
@@ -9,6 +9,28 @@

 #include <vector>

+torch::Tensor weak_ref_tensor(torch::Tensor& tensor) {
+  // Ensure tensor is on CUDA
+  if (!tensor.is_cuda()) {
+    throw std::runtime_error("Tensor must be on CUDA device");
+  }
+
+  // Get the raw data pointer
+  void* data_ptr = tensor.data_ptr();
+
+  // Get tensor sizes and strides
+  std::vector<int64_t> sizes = tensor.sizes().vec();
+  std::vector<int64_t> strides = tensor.strides().vec();
+
+  // Get tensor options (dtype, device)
+  auto options = tensor.options();
+
+  // Create a new tensor from the raw data pointer
+  auto new_tensor = torch::from_blob(data_ptr, sizes, strides, options);
+
+  return new_tensor;
+}
+
 // rms_norm and fused_add_rms_norm declarations also exist in
 // csrc/libtorch_stable/ops.h (torch::stable ABI for CUDA). They remain here
 // because the CPU build still uses these torch::Tensor declarations.
@@ -31,6 +53,16 @@ void silu_and_mul(torch::Tensor& out, torch::Tensor& input);
 void silu_and_mul_clamp(torch::Tensor& out, torch::Tensor& input, double limit,
                        double alpha = 1.0, double beta = 0.0);

+void silu_and_mul_quant(torch::Tensor& out, torch::Tensor& input,
+                        torch::Tensor& scale);
+
+void persistent_masked_m_silu_mul_quant(
+    const at::Tensor& input,   // (E, T, 2*H)
+    const at::Tensor& counts,  // (E)
+    at::Tensor& y_q,           // (E, T, H) [OUT]
+    at::Tensor& y_s,           // (E, T, H//group_size) [OUT]
+    bool use_ue8m0);
+
 void gelu_and_mul(torch::Tensor& out, torch::Tensor& input);

 void gelu_tanh_and_mul(torch::Tensor& out, torch::Tensor& input);
@@ -1,12 +1,16 @@
-#include "libtorch_stable/torch_utils.h"
+#include <ATen/cuda/CUDAContext.h>
+#include <torch/all.h>
+#include <c10/cuda/CUDAGuard.h>

 #include <cmath>
+#include "core/math.hpp"
+#include "../cuda_compat.h"
+#include "dispatch_utils.h"

-#include "libtorch_stable/core/math.hpp"
-#include "cuda_compat.h"
-#include "libtorch_stable/dispatch_utils.h"
 #include "quantization/w8a8/fp8/common.cuh"

+#include <c10/util/Float8_e4m3fn.h>
+
 #ifndef USE_ROCM
  #include <cuda_bf16.h>
  #include <cuda_fp16.h>
@@ -29,6 +33,7 @@ typedef __hip_fp8x4_e4m3_fnuz __nv_fp8x4_e4m3;
  #endif
 #endif

+#include "core/registration.h"
 namespace vllm {

 template <typename T>
@@ -559,47 +564,41 @@ __global__ void silu_mul_fp8_quant_deep_gemm_kernel(
 }  // namespace vllm

 // Launch activation, gating, and quantize kernel.
-#define LAUNCH_ACTIVATION_GATE_KERNEL(KERNEL)                             \
-  int d = input.size(-1) / 2;                                             \
-  int64_t num_tokens = input.numel() / input.size(-1);                    \
-  dim3 grid(num_tokens, num_tokens > 16 ? num_tokens > 32 ? 1 : 2 : 4);   \
-  dim3 block(std::min(d, 512));                                           \
-  const torch::stable::accelerator::DeviceGuard device_guard(             \
-      input.get_device_index());                                          \
-  const cudaStream_t stream =                                             \
-      get_current_cuda_stream(input.get_device_index());                  \
-  VLLM_STABLE_DISPATCH_FLOATING_TYPES(                                    \
-      input.scalar_type(), "act_and_mul_kernel", [&] {                    \
-        VLLM_STABLE_DISPATCH_FP8_TYPES(                                   \
-            out.scalar_type(), "act_and_mul_quant_kernel_fp8_type", [&] { \
-              vllm::act_and_mul_quant_kernel<scalar_t, KERNEL<scalar_t>,  \
-                                             fp8_t>                       \
-                  <<<grid, block, 0, stream>>>(                           \
-                      out.mutable_data_ptr<fp8_t>(),                      \
-                      input.const_data_ptr<scalar_t>(),                   \
-                      scale.const_data_ptr<float>(), d);                  \
-            });                                                           \
+#define LAUNCH_ACTIVATION_GATE_KERNEL(KERNEL)                               \
+  int d = input.size(-1) / 2;                                               \
+  int64_t num_tokens = input.numel() / input.size(-1);                      \
+  dim3 grid(num_tokens, num_tokens > 16 ? num_tokens > 32 ? 1 : 2 : 4);     \
+  dim3 block(std::min(d, 512));                                             \
+  const at::cuda::OptionalCUDAGuard device_guard(device_of(input));         \
+  const cudaStream_t stream = at::cuda::getCurrentCUDAStream();             \
+  VLLM_DISPATCH_FLOATING_TYPES(                                             \
+      input.scalar_type(), "act_and_mul_kernel", [&] {                      \
+        VLLM_DISPATCH_FP8_TYPES(                                            \
+            out.scalar_type(), "fused_add_rms_norm_kernel_fp8_type", [&] {  \
+              vllm::act_and_mul_quant_kernel<scalar_t, KERNEL<scalar_t>,    \
+                                             fp8_t>                         \
+                  <<<grid, block, 0, stream>>>(out.data_ptr<fp8_t>(),       \
+                                               input.data_ptr<scalar_t>(),  \
+                                               scale.data_ptr<float>(), d); \
+            });                                                             \
      });

-void silu_and_mul_quant(torch::stable::Tensor& out,    // [..., d]
-                        torch::stable::Tensor& input,  // [..., 2 * d]
-                        torch::stable::Tensor& scale) {
-  STD_TORCH_CHECK(
-      out.scalar_type() == torch::headeronly::ScalarType::Float8_e4m3fn ||
-      out.scalar_type() == torch::headeronly::ScalarType::Float8_e4m3fnuz);
-  STD_TORCH_CHECK(
-      input.scalar_type() == torch::headeronly::ScalarType::Half ||
-          input.scalar_type() == torch::headeronly::ScalarType::BFloat16,
-      "Input must be FP16 or BF16");
-  STD_TORCH_CHECK(input.size(-1) % 2 == 0);
+void silu_and_mul_quant(torch::Tensor& out,    // [..., d]
+                        torch::Tensor& input,  // [..., 2 * d]
+                        torch::Tensor& scale) {
+  TORCH_CHECK(out.dtype() == torch::kFloat8_e4m3fn ||
+              out.dtype() == torch::kFloat8_e4m3fnuz);
+  TORCH_CHECK(input.dtype() == torch::kFloat16 ||
+              input.dtype() == torch::kBFloat16);
+  TORCH_CHECK(input.size(-1) % 2 == 0);
  LAUNCH_ACTIVATION_GATE_KERNEL(vllm::silu_kernel);
 }

 void persistent_masked_m_silu_mul_quant(
-    const torch::stable::Tensor& input,              // (E, T, 2*H)
-    const torch::stable::Tensor& tokens_per_expert,  // (E)
-    torch::stable::Tensor& y_q,                      // (E, T, H) [OUT]
-    torch::stable::Tensor& y_s,  // (E, T, H//group_size) [OUT]
+    const at::Tensor& input,              // (E, T, 2*H)
+    const at::Tensor& tokens_per_expert,  // (E)
+    at::Tensor& y_q,                      // (E, T, H) [OUT]
+    at::Tensor& y_s,                      // (E, T, H//group_size) [OUT]
    bool cast_scale_ue8m0) {
 #ifndef USE_ROCM

@@ -607,18 +606,14 @@ void persistent_masked_m_silu_mul_quant(
  // fixed GROUP_SIZE of 128.
  static constexpr int GROUP_SIZE = 128;

-  STD_TORCH_CHECK(input.scalar_type() ==
-                  torch::headeronly::ScalarType::BFloat16);
-  STD_TORCH_CHECK(
-      y_q.scalar_type() == torch::headeronly::ScalarType::Float8_e4m3fn ||
-      y_q.scalar_type() == torch::headeronly::ScalarType::Float8_e4m3fnuz);
-  STD_TORCH_CHECK(input.size(-1) % (GROUP_SIZE * 2) == 0);
+  TORCH_CHECK(input.dtype() == torch::kBFloat16);
+  TORCH_CHECK(y_q.dtype() == torch::kFloat8_e4m3fn ||
+              y_q.dtype() == torch::kFloat8_e4m3fnuz);
+  TORCH_CHECK(input.size(-1) % (GROUP_SIZE * 2) == 0);

  bool const is_packed_ue8m0 =
-      (y_s.scalar_type() == torch::headeronly::ScalarType::Int &&
-       cast_scale_ue8m0);
-  STD_TORCH_CHECK(y_s.scalar_type() == torch::headeronly::ScalarType::Float ||
-                  is_packed_ue8m0);
+      (y_s.dtype() == torch::kInt32 && cast_scale_ue8m0);
+  TORCH_CHECK(y_s.dtype() == torch::kFloat32 || is_packed_ue8m0);

  using Idx_t = int64_t;

@@ -636,7 +631,7 @@ void persistent_masked_m_silu_mul_quant(

  int const NUM_GROUPS = H / GROUP_SIZE;

-  const cudaStream_t stream = get_current_cuda_stream(input.get_device_index());
+  const cudaStream_t stream = at::cuda::getCurrentCUDAStream();

  // TODO: Get this from cuda_arch ?
  static constexpr int SILU_V2_BLOCK_COUNT = 132 * 32;
@@ -648,21 +643,18 @@ void persistent_masked_m_silu_mul_quant(
    static constexpr int max_shared_mem_bytes =                                \
        GROUP_SIZE * 2 * STAGES * NUM_WARPS * 2;                               \
    dim3 grid(sms), block(THREAD_COUNT);                                       \
-    const torch::stable::accelerator::DeviceGuard device_guard(                \
-        input.get_device_index());                                             \
-    VLLM_STABLE_DISPATCH_FP8_TYPES(                                            \
+    const at::cuda::OptionalCUDAGuard device_guard(device_of(input));          \
+    VLLM_DISPATCH_FP8_TYPES(                                                   \
        y_q.scalar_type(), "silu_mul_fp8_quant_deep_gemm_kernel", [&] {        \
          vllm::silu_mul_fp8_quant_deep_gemm_kernel<                           \
              BLOCK_COUNT, max_shared_mem_bytes, fp8_t, scale_t, THREAD_COUNT, \
              Idx_t, CEIL_UE8M0, GROUP_SIZE, STAGES>                           \
              <<<grid, block, max_shared_mem_bytes + (E + 1) * 16, stream>>>(  \
-                  reinterpret_cast<const __nv_bfloat16*>(                      \
-                      input.const_data_ptr()),                                 \
-                  y_q.mutable_data_ptr<fp8_t>(),                               \
-                  reinterpret_cast<scale_t*>(y_s.mutable_data_ptr()),          \
-                  reinterpret_cast<const int32_t*>(                            \
-                      tokens_per_expert.const_data_ptr()),                     \
-                  E, T, H, stride_i_e, stride_i_t, stride_i_h, stride_yq_e,    \
+                  reinterpret_cast<__nv_bfloat16*>(input.data_ptr()),          \
+                  (fp8_t*)y_q.data_ptr(),                                      \
+                  reinterpret_cast<scale_t*>(y_s.data_ptr()),                  \
+                  reinterpret_cast<int32_t*>(tokens_per_expert.data_ptr()), E, \
+                  T, H, stride_i_e, stride_i_t, stride_i_h, stride_yq_e,       \
                  stride_yq_t, stride_yq_h, STRIDE_YS_E, STRIDE_YS_T,          \
                  STRIDE_YS_G, STRIDE_YS_P, stride_counts_e);                  \
        });
@@ -687,7 +679,7 @@ void persistent_masked_m_silu_mul_quant(
  Idx_t stride_ys_g = y_s.stride(2);
  Idx_t stride_ys_p = 0;
  if (!cast_scale_ue8m0) {
-    STD_TORCH_CHECK(!is_packed_ue8m0);
+    TORCH_CHECK(!is_packed_ue8m0);
    LAUNCH_ON_H(float, stride_ys_e, stride_ys_t, stride_ys_g, stride_ys_p,
                false);
    return;
@@ -700,8 +692,8 @@ void persistent_masked_m_silu_mul_quant(
    return;
  }

-  STD_TORCH_CHECK(cast_scale_ue8m0 && is_packed_ue8m0);
-  STD_TORCH_CHECK(y_s.scalar_type() == torch::headeronly::ScalarType::Int);
+  TORCH_CHECK(cast_scale_ue8m0 && is_packed_ue8m0);
+  TORCH_CHECK(y_s.dtype() == torch::kInt32);

  // Int32 packed ue8m0 scales tensor.
  // Let E, T, G be the number to experts, number of tokens and number of groups
@@ -1,5 +0,0 @@
-#include "core/registration.h"
-
-// QuTLASS registers torch.ops._qutlass_C via TORCH_LIBRARY in bindings.cpp.
-// This stub lets Python import vllm._qutlass_C to trigger op registration.
-REGISTER_EXTENSION(_qutlass_C)
@@ -20,6 +20,17 @@

 TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
  // vLLM custom ops
+  //
+
+  ops.def(
+      "persistent_masked_m_silu_mul_quant(Tensor input, Tensor counts, Tensor! "
+      "y_q, Tensor! y_s,"
+      "bool use_ue8m0) -> ()");
+  ops.impl("persistent_masked_m_silu_mul_quant", torch::kCUDA,
+           &persistent_masked_m_silu_mul_quant);
+
+  ops.def("weak_ref_tensor(Tensor input) -> Tensor");
+  ops.impl("weak_ref_tensor", torch::kCUDA, &weak_ref_tensor);

 #ifdef USE_ROCM
  // TODO: Remove this once we upgrade to torch 2.11.
@@ -28,6 +39,35 @@ TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
  ops.def("get_cuda_view_from_cpu_tensor(Tensor cpu_tensor) -> Tensor");
  ops.impl("get_cuda_view_from_cpu_tensor", torch::kCPU,
           &get_cuda_view_from_cpu_tensor);
+#endif
+
+  // Activation ops (quantized only — basic ops moved to _C_stable_libtorch)
+  ops.def(
+      "silu_and_mul_quant(Tensor! result, Tensor input, Tensor scale) -> ()");
+  ops.impl("silu_and_mul_quant", torch::kCUDA, &silu_and_mul_quant);
+
+  // Horizontally-fused DeepseekV4-MLA: per-head RMSNorm + GPT-J RoPE for Q, and
+  // GPT-J RoPE + UE8M0 FP8 quant + paged cache insert for KV, all in one
+  // kernel launch. Registered in _C_stable_libtorch (incl. the FlashInfer V4
+  // full-cache bf16/fp8 variants).
+
+  // Quantization ops
+#ifndef USE_ROCM
+
+  // Note about marlin kernel 'workspace' arguments:
+  // Technically these should be mutable since they are modified by the kernel.
+  // But since they are set back to zero once the kernel is finished we can
+  // hand wave and say that they have no net effect.
+  //
+  // The reason to mark 'workspace' as immutable is so that they don't interfere
+  // with using ScalarType arguments in the ops. If they are marked as mutable,
+  // pytorch throws an assert in
+  // 'torch._higher_order_ops._register_effectful_op' that prevents these
+  // kernels from being torch.compile'd.
+  // See the following document for more info on custom types and ops that use
+  // custom types:
+  // https://docs.google.com/document/d/18fBMPuOJ0fY5ZQ6YyrHUppw9FA332CpNtgB6SOIgyuA
+
 #endif
 }

@@ -75,13 +75,13 @@ RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRO
 # Install UMD
 RUN mkdir neo && \
    cd neo && \
-    wget https://github.com/intel/intel-graphics-compiler/releases/download/v2.34.4/intel-igc-core-2_2.34.4+21428_amd64.deb && \
-    wget https://github.com/intel/intel-graphics-compiler/releases/download/v2.34.4/intel-igc-opencl-2_2.34.4+21428_amd64.deb && \
-    wget https://github.com/intel/compute-runtime/releases/download/26.18.38308.1/intel-ocloc_26.18.38308.1-0_amd64.deb && \
-    wget https://github.com/intel/compute-runtime/releases/download/26.18.38308.1/intel-opencl-icd_26.18.38308.1-0_amd64.deb && \
-    wget https://github.com/intel/compute-runtime/releases/download/26.18.38308.1/libigdgmm12_22.10.0_amd64.deb && \
-    wget https://github.com/intel/compute-runtime/releases/download/26.18.38308.1/libze-intel-gpu1_26.18.38308.1-0_amd64.deb && \
-    wget https://github.com/oneapi-src/level-zero/releases/download/v1.28.2/level-zero_1.28.2+u24.04_amd64.deb && \
+    wget https://github.com/intel/intel-graphics-compiler/releases/download/v2.24.8/intel-igc-core-2_2.24.8+20344_amd64.deb && \
+    wget https://github.com/intel/intel-graphics-compiler/releases/download/v2.24.8/intel-igc-opencl-2_2.24.8+20344_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/intel-ocloc_25.48.36300.8-0_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/intel-opencl-icd_25.48.36300.8-0_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/libigdgmm12_22.8.2_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/libze-intel-gpu1_25.48.36300.8-0_amd64.deb && \
+    wget https://github.com/oneapi-src/level-zero/releases/download/v1.26.0/level-zero_1.26.0+u24.04_amd64.deb && \
    dpkg -i *.deb && \
    cd .. && \
    rm -rf neo
@@ -133,10 +133,10 @@ The model should inherit protocol `IsAttentionFree` and also implement class met
 For the mamba layers themselves, please use the [`MambaMixer`](../../../vllm/model_executor/layers/mamba/mamba_mixer.py) (for Mamba-1) or [`MambaMixer2`](../../../vllm/model_executor/layers/mamba/mamba_mixer2.py) (for Mamba-2) classes.
 The model should also be added to the `MODELS_CONFIG_MAP` dictionary in [vllm/model_executor/models/config.py](../../../vllm/model_executor/models/config.py) to ensure that the runtime defaults are optimized.

-For case (2), we recommend using as a reference the implementation of [`JambaForCausalLM`](../../../vllm/model_executor/models/jamba.py) (for an example of a model that uses Mamba-1 and attention together) or [`NemotronHForCausalLM`](../../../vllm/model_executor/models/nemotron_h.py) (for an example of a model that uses Mamba-2 and attention together).
+For case (2), we recommend using as a reference the implementation of [`JambaForCausalLM`](../../../vllm/model_executor/models/jamba.py) (for an example of a model that uses Mamba-1 and attention together) or [`BambaForCausalLM`](../../../vllm/model_executor/models/bamba.py) (for an example of a model that uses Mamba-2 and attention together).
 These models should follow the same instructions as case (1), but they should inherit protocol `IsHybrid` (instead of `IsAttentionFree`) and it is *not* necessary to add them to the `MODELS_CONFIG_MAP` (their runtime defaults will be inferred from the protocol).

-For case (3), we recommend looking at the implementation of [`Lfm2ForCausalLM`](../../../vllm/model_executor/models/lfm2.py) as a reference, which uses a custom "mamba-like" layer `ShortConv`.
+For case (3), we recommend looking at the implementation of [`MiniMaxText01ForCausalLM`](../../../vllm/model_executor/models/minimax_text_01.py) or [`Lfm2ForCausalLM`](../../../vllm/model_executor/models/lfm2.py) as a reference, which use custom "mamba-like" layers `MiniMaxText01LinearAttention` and `ShortConv` respectively.
 Please follow the same guidelines as case (2) for implementing these models.
 We use "mamba-like" to refer to layers that possess a state that is updated in-place, rather than being appended-to (like KV cache for attention).
 For implementing new custom mamba-like layers, one should inherit from `MambaBase` and implement the methods `get_state_dtype`, `get_state_shape` to calculate the data types and state shapes at runtime, as well as `mamba_type` and `get_attn_backend`.
@@ -144,5 +144,5 @@ It is also necessary to implement the "attention meta-data" class which handles
 Please see [`LinearAttentionMetadata`](../../../vllm/v1/attention/backends/linear_attn.py) or [`ShortConvAttentionMetadata`](../../../vllm/v1/attention/backends/short_conv_attn.py) for examples of this.
 It is also worth noting that we should update `MambaAttentionBackendEnum` in [`registry.py`](../../../vllm/v1/attention/backends/registry.py) when adding a new mamba backend.
 Finally, if one wants to support torch compile and CUDA graphs, it necessary to wrap the call to the mamba-like layer inside a custom op and register it.
-Please see the calls to `direct_register_custom_op` in [vllm/model_executor/layers/mamba/linear/minimax_linear_attn.py](../../../vllm/model_executor/layers/mamba/linear/minimax_linear_attn.py) or [vllm/model_executor/layers/mamba/short_conv.py](../../../vllm/model_executor/layers/mamba/short_conv.py) for examples of this.
+Please see the calls to `direct_register_custom_op` in [vllm/model_executor/models/minimax_text_01.py](../../../vllm/model_executor/models/minimax_text_01.py) or [vllm/model_executor/layers/mamba/short_conv.py](../../../vllm/model_executor/layers/mamba/short_conv.py) for examples of this.
 The new custom op should then be added to the list `_attention_ops` in [vllm/config/compilation.py](../../../vllm/config/compilation.py) to ensure that piecewise CUDA graphs works as intended.
@@ -4,7 +4,7 @@ Deploying vLLM on Kubernetes is a scalable and efficient way to serve machine le

 * **Upstream vLLM compatibility** – It wraps around upstream vLLM without modifying its code.
 * **Ease of use** – Simplified deployment via Helm charts and observability through Grafana dashboards.
-* **High performance** – Optimized for LLM workloads with features like multimodel support, model-aware and prefix-aware routing, fast vLLM bootstrapping, and KV cache offloading with [LMCache](https://github.com/LMCache/LMCache) (wired up in vLLM via `--kv-offloading-backend lmcache`; see the [LMCache examples](https://github.com/vllm-project/vllm/tree/main/examples/disaggregated/lmcache) and [docs.lmcache.ai](https://docs.lmcache.ai)), among others.
+* **High performance** – Optimized for LLM workloads with features like multimodel support, model-aware and prefix-aware routing, fast vLLM bootstrapping, and KV cache offloading with [LMCache](https://github.com/LMCache/LMCache), among others.

 If you are new to Kubernetes, don't worry: in the vLLM production stack [repo](https://github.com/vllm-project/production-stack), we provide a step-by-step [guide](https://github.com/vllm-project/production-stack/blob/main/tutorials/00-install-kubernetes-env.md) and a [short video](https://www.youtube.com/watch?v=EsTJbQtzj0g) to set up everything and get started in **4 minutes**!

@@ -170,8 +170,8 @@ Priority is **1 = highest** (tried first).
 | Backend | Version | Dtypes | KV Dtypes | Block Sizes | Head Sizes | Sink | Non-Causal | MM Prefix | DCP | Attention Types | Compute Cap. |
 | ------- | ------- | ------ | --------- | ----------- | ---------- | ---- | ---------- | --------- | --- | --------------- | ------------ |
 | `CPU_ATTN` | | fp16, bf16, fp32 | `auto`, `fp8`, `fp8_e4m3`, `fp8_e5m2` | %16 | 32, 64, 80, 96, 112, 128, 160, 192, 224, 256, 512 | ❌ | ❌ | ❌ | ❌ | All | N/A |
-| `FLASHINFER` | Native† | fp16, bf16 | `auto`, `float16`, `bfloat16`, `fp8`, `fp8_e4m3`, `fp8_e5m2` | 16, 32, 64, 128, 256, 512, 1024 | 64, 128, 256, 512 | ❌ | ✅ | ❌ | ✅ | Decoder | 7.x-9.x |
-| `FLASHINFER` | TRTLLM† | fp16, bf16 | `auto`, `float16`, `bfloat16`, `fp8`, `fp8_e4m3`, `fp8_e5m2`, `nvfp4` | 16, 32, 64, 128, 256, 512, 1024 | 64, 128, 256, 512 | ✅ | ✅ | ❌ | ✅ | Decoder | 10.x |
+| `FLASHINFER` | Native† | fp16, bf16 | `auto`, `float16`, `bfloat16`, `fp8`, `fp8_e4m3`, `fp8_e5m2` | 16, 32, 64, 128, 256, 512, 1024 | 64, 128, 256, 512 | ❌ | ❌ | ❌ | ✅ | Decoder | 7.x-9.x |
+| `FLASHINFER` | TRTLLM† | fp16, bf16 | `auto`, `float16`, `bfloat16`, `fp8`, `fp8_e4m3`, `fp8_e5m2`, `nvfp4` | 16, 32, 64, 128, 256, 512, 1024 | 64, 128, 256, 512 | ✅ | ❌ | ❌ | ✅ | Decoder | 10.x |
 | `FLASH_ATTN` | FA2* | fp16, bf16 | `auto`, `float16`, `bfloat16` | %16 | Any | ❌ | ✅ | ❌ | ✅ | All | ≥8.0 |
 | `FLASH_ATTN` | FA3* | fp16, bf16 | `auto`, `float16`, `bfloat16`, `fp8`, `fp8_e4m3`, `fp8_e5m2` | %16 | Any | ✅ | ✅ | ❌ | ✅ | All | 9.x |
 | `FLASH_ATTN` | FA4* | fp16, bf16 | `auto`, `float16`, `bfloat16` | %16 | Any | ✅ | ✅ | ❌ | ✅ | All | ≥10.0 |
@@ -20,7 +20,7 @@ Two main reasons:
 Now supports 9 types of connectors:

 - **ExampleConnector**: refer to [examples/disaggregated/example_connector/run.sh](../../examples/disaggregated/example_connector/run.sh) for the example usage of ExampleConnector disaggregated prefilling.
- **LMCacheConnectorV1**: refer to [examples/disaggregated/lmcache/disagg_prefill_lmcache_v1/disagg_example_nixl.sh](../../examples/disaggregated/lmcache/disagg_prefill_lmcache_v1/disagg_example_nixl.sh) for the example usage of LMCacheConnectorV1 disaggregated prefilling which uses NIXL as the underlying KV transmission. LMCache also offers a multi-process (MP) mode via `LMCacheMPConnector`, where a standalone `lmcache server` holds the KV cache shared by one or more vLLM instances; see the [LMCache examples](../../examples/disaggregated/lmcache/README.md) and the [LMCache docs](https://docs.lmcache.ai) for setup.
+- **LMCacheConnectorV1**: refer to [examples/disaggregated/lmcache/disagg_prefill_lmcache_v1/disagg_example_nixl.sh](../../examples/disaggregated/lmcache/disagg_prefill_lmcache_v1/disagg_example_nixl.sh) for the example usage of LMCacheConnectorV1 disaggregated prefilling which uses NIXL as the underlying KV transmission.
 - **NixlConnector**: refer to [tests/v1/kv_connector/nixl_integration/run_accuracy_test.sh](../../tests/v1/kv_connector/nixl_integration/run_accuracy_test.sh) for the example usage of NixlConnector disaggregated prefilling which support fully async send/recv. For detailed usage guide, see [NixlConnector Usage Guide](nixl_connector_usage.md). For feature compatibility details, see [NixlConnector Compatibility Matrix](nixl_connector_compatibility.md). You may specify one or multiple NIXL transfer backends, such as:

  ```bash
@@ -74,7 +74,6 @@ vllm serve <model> \
 | `max_tracker_size` | no | `64000` | single-tier | Max entries in the lookup tracker. |
 | `secondary_tiers` | no | `[]` | multi-tier | List of secondary tier configs (see below). |
 | `offload_prompt_only` | no | `true` | both | If `true`, only prompt (prefill) blocks are offloaded; decode blocks are skipped. |
-| `self_describing_kv_events` | no | `false` | single-tier | Opt-in. When `true` *and* KV cache events are enabled (`--kv-events-config` with `enable_kv_cache_events`), the connector emits self-describing block-granular `BlockStored`/`BlockRemoved` payloads (constituent block hashes, whole-chunk `token_ids`, per-block `block_size`, parent hash, LoRA + group/cache-spec metadata) instead of the placeholder fallback, so external KV-event consumers can index offloaded blocks. Inert unless events are enabled. Currently rejected by `TieringOffloadingSpec`. Full-attention groups only; sliding-window/SSM groups keep the placeholder fallback. In chunk mode (`block_size` > GPU block size), overlapping chunks re-announce shared per-block hashes, so consumers must reference-count (deduplicate) repeated store/remove announcements. |
 | `spec_module_path` | no | — | both | Python import path for a custom `OffloadingSpec` not in the built-in registry. Required only when `spec_name` is not built-in (advanced). |

 ## Secondary Tiers
@@ -203,7 +203,6 @@ the vLLM JSON config.
 ### kv_connector_extra_config

 - `load_async` (bool): Enable asynchronous loading for better compute-I/O overlap. Default: `true`.
- `lookup_async` (bool): Run the external prefix-cache lookup on a background thread so it never blocks the scheduler step. The request is held until the in-flight lookup completes, then resumed on a later step. Default: `false`.
 - `enable_cross_layers_blocks` (bool): Enable cross-layer block packing for reduced store operations. Default: `false`.
 - `lookup_rpc_port` (int): Custom port for the ZMQ lookup RPC socket. Default: `0`.
 - `cache_prefix` (str): Namespace prepended to every store key. Lets separate deployments share one Mooncake master without polluting each other — instances configured with different prefixes never see each other's cached blocks, even for identical prompts. All instances that should share a prefix cache must use the same value. Default: `""` (no prefix; keys are byte-identical to the unprefixed format).
@@ -321,6 +321,15 @@ For Qwen2.5, the chat template in tokenizer_config.json has already included sup

 Flags: `--tool-call-parser hermes`

+### MiniMax Models (`minimax_m1`)
+
+Supported models:
+
+* `MiniMaxAi/MiniMax-M1-40k` (use with [examples/tool_chat_template_minimax_m1.jinja](../../examples/tool_chat_template_minimax_m1.jinja))
+* `MiniMaxAi/MiniMax-M1-80k` (use with [examples/tool_chat_template_minimax_m1.jinja](../../examples/tool_chat_template_minimax_m1.jinja))
+
+Flags: `--tool-call-parser minimax --chat-template examples/tool_chat_template_minimax_m1.jinja`
+
 ### DeepSeek-V3 Models (`deepseek_v3`)

 Supported models:
@@ -27,7 +27,6 @@ Currently, there are no pre-built XPU wheels.

 - First, install required [driver](https://dgpu-docs.intel.com/driver/installation.html#installing-gpu-drivers).
 - Second, install Python packages for vLLM XPU backend building (Intel OneAPI dependencies are installed automatically as part of `torch-xpu`, see [PyTorch XPU get started](https://docs.pytorch.org/docs/stable/notes/get_start_xpu.html)):
- Start from vllm-xpu-kernels v0.1.10, we recommend user upgrade driver to [compute runtime 26.18](https://github.com/intel/compute-runtime/releases/tag/26.14.37833.4) release, to avoid potential compatibility issue.

 ```bash
 git clone https://github.com/vllm-project/vllm.git
@@ -61,7 +61,7 @@ Models of any architecture can be converted into embedding models using `--conve
 | `ColModernVBertForRetrieval` | ColModernVBERT | T / I | `ModernVBERT/colmodernvbert-merged` | | |
 | `ColPaliForRetrieval` | ColPali | T / I | `vidore/colpali-v1.3-hf` | | |
 | `ColQwen3` | Qwen3-VL | T / I | `TomoroAI/tomoro-colqwen3-embed-4b`, `TomoroAI/tomoro-colqwen3-embed-8b` | | |
-| `ColQwen3_5` | ColQwen3.5 | T + I + V | `athrael-soju/colqwen3.5-4.5B-v3`, `vultr/VultronRetrieverPrime-Qwen3.5-8B` | | |
+| `ColQwen3_5` | ColQwen3.5 | T + I + V | `athrael-soju/colqwen3.5-4.5B-v3` | | |
 | `OpsColQwen3Model` | Qwen3-VL | T / I | `OpenSearch-AI/Ops-Colqwen3-4B`, `OpenSearch-AI/Ops-Colqwen3-8B` | | |
 | `Qwen3VLNemotronEmbedModel` | Qwen3-VL | T / I | `nvidia/nemotron-colembed-vl-4b-v2`, `nvidia/nemotron-colembed-vl-8b-v2` | ✅︎ | ✅︎ |
 | `*ForConditionalGeneration`<sup>C</sup>, `*ForCausalLM`<sup>C</sup>, etc. | Generative models | \* | N/A | \* | \* |
@@ -441,6 +441,7 @@ th {
 | `MiMoV2ForCausalLM` | MiMoV2Pro | `XiaomiMiMo/MiMo-V2.5-Pro`, etc. | | ✅︎ |
 | `MiniCPMForCausalLM` | MiniCPM | `openbmb/MiniCPM-2B-sft-bf16`, `openbmb/MiniCPM-2B-dpo-bf16`, `openbmb/MiniCPM-S-1B-sft`, etc. | ✅︎ | ✅︎ |
 | `MiniCPM3ForCausalLM` | MiniCPM3 | `openbmb/MiniCPM3-4B`, etc. | ✅︎ | ✅︎ |
+| `MiniMaxForCausalLM` | MiniMax-Text | `MiniMaxAI/MiniMax-Text-01-hf`, etc. | | |
 | `MiniMaxM2ForCausalLM` | MiniMax-M2, MiniMax-M2.1 | `MiniMaxAI/MiniMax-M2`, etc. | ✅︎ | ✅︎ |
 | `MistralForCausalLM` | Ministral-3, Mistral, Mistral-Instruct | `mistralai/Ministral-3-3B-Instruct-2512`, `mistralai/Mistral-7B-v0.1`, `mistralai/Mistral-7B-Instruct-v0.1`, etc. | ✅︎ | ✅︎ |
 | `MistralLarge3ForCausalLM` | Mistral-Large-3-675B-Base-2512, Mistral-Large-3-675B-Instruct-2512 | `mistralai/Mistral-Large-3-675B-Base-2512`, `mistralai/Mistral-Large-3-675B-Instruct-2512`, etc. | ✅︎ | ✅︎ |
@@ -486,6 +487,8 @@ th {
 | `TeleChat2ForCausalLM` | TeleChat2 | `Tele-AI/TeleChat2-3B`, `Tele-AI/TeleChat2-7B`, `Tele-AI/TeleChat2-35B`, etc. | ✅︎ | ✅︎ |
 | `TeleChat3ForCausalLM` | TeleChat3 | `Tele-AI/TeleChat3-36B-Thinking`, `Tele-AI/TeleChat3-Coder-36B-Thinking`, etc. | ✅︎ | ✅︎ |
 | `TeleFLMForCausalLM` | TeleFLM | `CofeAI/FLM-2-52B-Instruct-2407`, `CofeAI/Tele-FLM`, etc. | ✅︎ | ✅︎ |
+| `MiniMaxM1ForCausalLM` | MiniMax-Text | `MiniMaxAI/MiniMax-M1-40k`, `MiniMaxAI/MiniMax-M1-80k`, etc. | | |
+| `MiniMaxText01ForCausalLM` | MiniMax-Text | `MiniMaxAI/MiniMax-Text-01`, etc. | | |
 | `Zamba2ForCausalLM` | Zamba2 | `Zyphra/Zamba2-7B-instruct`, `Zyphra/Zamba2-2.7B-instruct`, `Zyphra/Zamba2-1.2B-instruct`, etc. | | |

 !!! note
@@ -592,7 +595,6 @@ These models primarily accept the [`LLM.generate`](./generative_models.md#llmgen
 | `MiMoV2OmniForCausalLM` | MiMo-V2.5-Omni | T + I<sup>E+</sup> + V<sup>E+</sup> + A<sup>+</sup> | `XiaomiMiMo/MiMo-V2.5-Omni` | | ✅︎ |
 | `MiniCPMO` | MiniCPM-O | T + I<sup>E+</sup> + V<sup>E+</sup> + A<sup>E+</sup> | `openbmb/MiniCPM-o-2_6`, etc. | ✅︎ | ✅︎ |
 | `MiniCPMV` | MiniCPM-V | T + I<sup>E+</sup> + V<sup>E+</sup> | `openbmb/MiniCPM-V-2` (see note), `openbmb/MiniCPM-Llama3-V-2_5`, `openbmb/MiniCPM-V-2_6`, `openbmb/MiniCPM-V-4`, `openbmb/MiniCPM-V-4_5`, etc. | ✅︎ | |
-| `MiniMaxM3SparseForConditionalGeneration` | MiniMax-M3 | T + I<sup>+</sup> + V<sup>+</sup> | `MiniMaxAI/MiniMax-M3`, `MiniMaxAI/MiniMax-M3-MXFP8`, etc. | | |
 | `MiniMaxVL01ForConditionalGeneration` | MiniMax-VL | T + I<sup>E+</sup> | `MiniMaxAI/MiniMax-VL-01`, etc. | | ✅︎ |
 | `Mistral3ForConditionalGeneration` | Mistral3 (HF Transformers) | T + I<sup>+</sup> | `mistralai/Mistral-Small-3.1-24B-Instruct-2503`, etc. | ✅︎ | ✅︎ |
 | `MolmoForCausalLM` | Molmo | T + I<sup>+</sup> | `allenai/Molmo-7B-D-0924`, `allenai/Molmo-7B-O-0924`, etc. | ✅︎ | ✅︎ |
@@ -128,7 +128,7 @@ Models that use Mamba-2 and Mamba-1 layers (e.g., `Mamba2ForCausalLM`, `MambaFor
 Hybrid models that combine Mamba-2 and Mamba-1 layers with standard attention layers are also supported (e.g., `BambaForCausalLM`,
 `Zamba2ForCausalLM`, `NemotronHForCausalLM`, `FalconH1ForCausalLM` and `GraniteMoeHybridForCausalLM`, `JambaForCausalLM`, `Plamo2ForCausalLM`).

-Hybrid models with mechanisms different to Mamba are also supported (e.g, `Lfm2ForCausalLM`).
+Hybrid models with mechanisms different to Mamba are also supported (e.g, `MiniMaxText01ForCausalLM`, `MiniMaxM1ForCausalLM`, `Lfm2ForCausalLM`).

 Please note that prefix caching is not yet supported for any of the above models.

@@ -1,38 +1,10 @@
 # LMCache Examples

-This folder demonstrates how to use LMCache with vLLM v1 for KV cache
-offloading, disaggregated prefilling, and KV cache sharing.
+This folder demonstrates how to use LMCache for disaggregated prefilling, CPU offloading and KV cache sharing.

-## Integration modes
+## 1. Disaggregated Prefill in vLLM v1

-LMCache integrates with vLLM v1 in two ways:
-
- **In-process mode** (`LMCacheConnectorV1`): LMCache runs inside the vLLM
-  process and is configured through environment variables or a YAML config
-  file (`LMCACHE_CONFIG_FILE`). This is the simplest way to add single-node
-  CPU/disk offloading.
- **Multi-process (MP) mode** (`LMCacheMPConnector`): LMCache runs as a
-  standalone server (`lmcache server`) that owns the KV cache storage; one or
-  more vLLM instances connect to it. This is the recommended mode for
-  distributed KV storage and for sharing KV cache across instances. See the
-  [LMCache docs](https://docs.lmcache.ai) for the full MP setup.
-
-## 1. CPU offload (in-process)
-
- `python cpu_offload_lmcache.py` - CPU offloading with `LMCacheConnectorV1`
-  for vLLM v1.
-
-## 2. CPU offload (multi-process)
-
- `bash cpu_offload_lmcache_mp.sh` - CPU offloading with `LMCacheMPConnector`,
-  using a standalone `lmcache server`. vLLM provides a built-in shortcut for
-  this setup via `--kv-offloading-backend lmcache` and
-  `--kv-offloading-size <GiB>`.
-
-## 3. Disaggregated Prefill in vLLM v1
-
-This example demonstrates how to run LMCache with disaggregated prefill using
-NIXL on a single node.
+This example demonstrates how to run LMCache with disaggregated prefill using NIXL on a single node.

 ### Prerequisites

@@ -74,7 +46,15 @@ The main script generates several log files:
 - `decoder.log` - Logs from the decode server
 - `proxy.log` - Logs from the proxy server

-## 4. KV Cache Sharing
+## 2. CPU Offload Examples

-The `kv_cache_sharing_lmcache_v1.py` example demonstrates how to share KV
-caches between vLLM v1 instances through a centralized LMCache server.
+- `python cpu_offload_lmcache.py -v v0` - CPU offloading implementation for vLLM v0
+- `python cpu_offload_lmcache.py -v v1` - CPU offloading implementation for vLLM v1
+
+## 3. KV Cache Sharing
+
+The `kv_cache_sharing_lmcache_v1.py` example demonstrates how to share KV caches between vLLM v1 instances.
+
+## 4. Disaggregated Prefill in vLLM v0
+
+The `disaggregated_prefill_lmcache_v0.py` provides an example of how to run disaggregated prefill in vLLM v0.
@@ -1,8 +1,20 @@
 # SPDX-License-Identifier: Apache-2.0
 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
 """
-This file demonstrates the example usage of CPU offloading
-with LMCache in vLLM v1.
+This file demonstrates the example usage of cpu offloading
+with LMCache in vLLM v1 or v0.
+
+Usage:
+
+    Specify vLLM version
+
+    -v v0 : Use LMCacheConnector
+            model = mistralai/Mistral-7B-Instruct-v0.2
+            (Includes enable_chunked_prefill = True)
+
+    -v v1 : Use LMCacheConnectorV1 (default)
+            model = meta-llama/Meta-Llama-3.1-8B-Instruct
+            (Without enable_chunked_prefill)

 Note that `lmcache` is needed to run this example.
 Requirements:
@@ -11,6 +23,7 @@ Learn more about LMCache environment setup, please refer to:
 https://docs.lmcache.ai/getting_started/installation.html
 """

+import argparse
 import contextlib
 import os
 import time
@@ -26,6 +39,8 @@ from vllm.engine.arg_utils import EngineArgs

 def setup_environment_variables():
    # LMCache-related environment variables
+    # Use experimental features in LMCache
+    os.environ["LMCACHE_USE_EXPERIMENTAL"] = "True"
    # LMCache is set to use 256 tokens per chunk
    os.environ["LMCACHE_CHUNK_SIZE"] = "256"
    # Enable local CPU backend in LMCache
@@ -35,9 +50,9 @@ def setup_environment_variables():


@contextlib.contextmanager
-def build_llm_with_lmcache(model: str):
+def build_llm_with_lmcache(lmcache_connector: str, model: str):
    ktc = KVTransferConfig(
-        kv_connector="LMCacheConnectorV1",
+        kv_connector=lmcache_connector,
        kv_role="kv_both",
    )
    # Set GPU memory utilization to 0.8 for an A40 GPU with 40GB
@@ -77,10 +92,23 @@ def print_output(
    print("-" * 50)


+def parse_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-v",
+        "--version",
+        choices=["v0", "v1"],
+        default="v1",
+        help="Specify vLLM version (default: v1)",
+    )
+    return parser.parse_args()
+
+
 def main():
+    lmcache_connector = "LMCacheConnectorV1"
    model = "meta-llama/Meta-Llama-3.1-8B-Instruct"
    setup_environment_variables()
-    with build_llm_with_lmcache(model) as llm:
+    with build_llm_with_lmcache(lmcache_connector, model) as llm:
        # This example script runs two requests with a shared prefix.
        # Define the shared prompt and specific prompts
        shared_prompt = "Hello, how are you?" * 1000
@@ -1,43 +0,0 @@
-#!/bin/bash
-# SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
-#
-# CPU offloading with LMCache in multi-process (MP) mode.
-#
-# In MP mode, LMCache runs as a standalone server process (`lmcache server`)
-# that owns the KV cache storage. One or more vLLM instances connect to it via
-# the `LMCacheMPConnector`. This is the recommended way to run LMCache for
-# distributed KV storage and for sharing KV cache across vLLM instances.
-#
-# vLLM ships a built-in shortcut for this setup: pass `--kv-offloading-backend
-# lmcache` together with `--kv-offloading-size <GiB>` and vLLM wires up the
-# `LMCacheMPConnector` for you (it defaults to the LMCache server at
-# tcp://localhost:5555, matching the `lmcache server` default).
-#
-# Requires `lmcache` to be installed (`pip install lmcache`).
-# Learn more: https://docs.lmcache.ai
-set -euo pipefail
-
-MODEL=${MODEL:-meta-llama/Meta-Llama-3.1-8B-Instruct}
-
-# 1. Launch the standalone LMCache server (binds tcp://localhost:5555 by
-#    default). `--l1-size-gb` sets the CPU memory budget for the L1 cache.
-echo "Starting LMCache server..."
-lmcache server --host localhost --port 5555 --l1-size-gb 5 &
-LMCACHE_SERVER_PID=$!
-trap 'kill $LMCACHE_SERVER_PID 2>/dev/null || true' EXIT
-
-# 2. Launch vLLM and offload KV cache to the LMCache server.
-#    The MP connector currently requires the non-hybrid KV cache manager.
-echo "Starting vLLM server with LMCache MP offloading..."
-vllm serve "$MODEL" \
-    --port 8000 \
-    --kv-offloading-size 5 \
-    --kv-offloading-backend lmcache \
-    --disable-hybrid-kv-cache-manager
-
-# Equivalent explicit configuration (instead of the two flags above):
-#   --kv-transfer-config \
-#   '{"kv_connector":"LMCacheMPConnector","kv_role":"kv_both",
-#     "kv_connector_extra_config":{"lmcache.mp.host":"tcp://localhost",
-#                                  "lmcache.mp.port":5555}}'
@@ -0,0 +1,144 @@
+# SPDX-License-Identifier: Apache-2.0
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
+"""
+This file demonstrates the example usage of disaggregated prefilling
+with LMCache.
+We will launch 2 vllm instances (GPU 0 for prefill and GPU 1 for decode),
+and launch an additional LMCache server.
+KV cache is transferred in the following manner:
+vLLM prefill node -> LMCache server -> vLLM decode node.
+
+Note that `pip install lmcache` is needed to run this example.
+Learn more about LMCache in https://github.com/LMCache/LMCache.
+"""
+
+import os
+import subprocess
+import time
+from multiprocessing import Event, Process
+
+from lmcache.experimental.cache_engine import LMCacheEngineBuilder
+from lmcache.integration.vllm.utils import ENGINE_NAME
+
+from vllm import LLM, SamplingParams
+from vllm.config import KVTransferConfig
+
+# LMCache-related environment variables
+# The port to start LMCache server
+port = 8100
+# Use experimental features in LMCache
+os.environ["LMCACHE_USE_EXPERIMENTAL"] = "True"
+# LMCache is set to use 256 tokens per chunk
+os.environ["LMCACHE_CHUNK_SIZE"] = "256"
+# Disable local CPU backend in LMCache
+os.environ["LMCACHE_LOCAL_CPU"] = "False"
+# Set local CPU memory buffer limit to 5.0 GB
+os.environ["LMCACHE_MAX_LOCAL_CPU_SIZE"] = "5.0"
+# Set the remote URL for LMCache server
+os.environ["LMCACHE_REMOTE_URL"] = f"lm://localhost:{port}"
+# Set the serializer/deserializer between vllm and LMCache server
+# `naive` indicates using raw bytes of the tensor without any compression
+os.environ["LMCACHE_REMOTE_SERDE"] = "naive"
+
+prompts = [
+    "Hello, how are you?" * 1000,
+]
+
+
+def run_prefill(prefill_done, prompts):
+    # We use GPU 0 for prefill node.
+    os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+
+    sampling_params = SamplingParams(temperature=0, top_p=0.95, max_tokens=1)
+
+    ktc = KVTransferConfig(
+        kv_connector="LMCacheConnector",
+        kv_role="kv_producer",
+        kv_rank=0,
+        kv_parallel_size=2,
+    )
+    # Set GPU memory utilization to 0.8 for an A40 GPU with 40GB
+    # memory. Reduce the value if your GPU has less memory.
+    llm = LLM(
+        model="mistralai/Mistral-7B-Instruct-v0.2",
+        kv_transfer_config=ktc,
+        max_model_len=8000,
+        gpu_memory_utilization=0.8,
+        enforce_eager=True,
+    )
+
+    # llm.generate(prompts, sampling_params)
+    outputs = llm.generate(prompts, sampling_params)
+    for output in outputs:
+        generated_text = output.outputs[0].text
+        print(f"Generated text: {generated_text!r}")
+    print("Prefill node is finished.")
+    prefill_done.set()
+
+    # Clean up lmcache backend
+    LMCacheEngineBuilder.destroy(ENGINE_NAME)
+
+
+def run_decode(prefill_done, prompts, timeout=1):
+    # We use GPU 1 for decode node.
+    os.environ["CUDA_VISIBLE_DEVICES"] = "1"
+
+    sampling_params = SamplingParams(temperature=0, top_p=0.95, max_tokens=10)
+
+    ktc = KVTransferConfig(
+        kv_connector="LMCacheConnector",
+        kv_role="kv_consumer",
+        kv_rank=1,
+        kv_parallel_size=2,
+    )
+    # Set GPU memory utilization to 0.8 for an A40 GPU with 40GB
+    # of memory. Reduce the value if your GPU has less memory.
+    llm = LLM(
+        model="mistralai/Mistral-7B-Instruct-v0.2",
+        kv_transfer_config=ktc,
+        max_model_len=8000,
+        gpu_memory_utilization=0.8,
+        enforce_eager=True,
+    )
+
+    print("Waiting for prefill node to finish...")
+    prefill_done.wait()
+    time.sleep(timeout)
+
+    outputs = llm.generate(prompts, sampling_params)
+    for output in outputs:
+        generated_text = output.outputs[0].text
+        print(f"Generated text: {generated_text!r}")
+
+    # Clean up lmcache backend
+    LMCacheEngineBuilder.destroy(ENGINE_NAME)
+
+
+def run_lmcache_server(port):
+    server_proc = subprocess.Popen(
+        ["python", "-m", "lmcache.experimental.server", "localhost", str(port)]
+    )
+    return server_proc
+
+
+def main():
+    prefill_done = Event()
+    prefill_process = Process(target=run_prefill, args=(prefill_done, prompts))
+    decode_process = Process(target=run_decode, args=(prefill_done, prompts))
+    lmcache_server_process = run_lmcache_server(port)
+
+    # Start prefill node
+    prefill_process.start()
+
+    # Start decode node
+    decode_process.start()
+
+    # Clean up the processes
+    decode_process.join()
+    prefill_process.terminate()
+    lmcache_server_process.terminate()
+    lmcache_server_process.wait()
+
+
+if __name__ == "__main__":
+    main()
@@ -30,6 +30,7 @@ if [[ $1 == "prefiller" ]]; then

    UCX_TLS=cuda_ipc,cuda_copy,tcp \
        LMCACHE_CONFIG_FILE=$prefill_config_file \
+        LMCACHE_USE_EXPERIMENTAL=True \
        VLLM_ENABLE_V1_MULTIPROCESSING=1 \
        VLLM_WORKER_MULTIPROC_METHOD=spawn \
        CUDA_VISIBLE_DEVICES=0 \
@@ -46,6 +47,7 @@ elif [[ $1 == "decoder" ]]; then

    UCX_TLS=cuda_ipc,cuda_copy,tcp \
        LMCACHE_CONFIG_FILE=$decode_config_file \
+        LMCACHE_USE_EXPERIMENTAL=True \
        VLLM_ENABLE_V1_MULTIPROCESSING=1 \
        VLLM_WORKER_MULTIPROC_METHOD=spawn \
        CUDA_VISIBLE_DEVICES=1 \
@@ -26,6 +26,8 @@ from vllm.config import KVTransferConfig
 # LMCache-related environment variables
 # The port to start LMCache server
 port = 8100
+# Use experimental features in LMCache
+os.environ["LMCACHE_USE_EXPERIMENTAL"] = "True"
 # LMCache is set to use 256 tokens per chunk
 os.environ["LMCACHE_CHUNK_SIZE"] = "256"
 # Disable local CPU backend in LMCache
@@ -1481,6 +1481,39 @@ def run_minicpmv(questions: list[str], modality: str) -> ModelRequestData:
    return run_minicpmv_base(questions, modality, "openbmb/MiniCPM-V-2_6")


+def run_minimax_vl_01(questions: list[str], modality: str) -> ModelRequestData:
+    assert modality == "image"
+
+    model_name = "MiniMaxAI/MiniMax-VL-01"
+
+    engine_args = EngineArgs(
+        model=model_name,
+        max_num_seqs=2,
+        limit_mm_per_prompt={modality: 1},
+        trust_remote_code=True,
+        tensor_parallel_size=8,
+    )
+
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    messages = [
+        [
+            {
+                "role": "user",
+                "content": [{"type": "image"}, {"type": "text", "text": question}],
+            }
+        ]
+        for question in questions
+    ]
+    prompts = tokenizer.apply_chat_template(
+        messages, add_generation_prompt=True, tokenize=False
+    )
+
+    return ModelRequestData(
+        engine_args=engine_args,
+        prompts=prompts,
+    )
+
+
 # Mistral-3 HF-format
 def run_mistral3(questions: list[str], modality: str) -> ModelRequestData:
    assert modality == "image"
@@ -2452,6 +2485,7 @@ model_example_map = {
    "mantis": run_mantis,
    "minicpmo": run_minicpmo,
    "minicpmv": run_minicpmv,
+    "minimax_vl_01": run_minimax_vl_01,
    "mistral3": run_mistral3,
    "molmo": run_molmo,
    "molmo2": run_molmo2,
@@ -7,27 +7,11 @@ ColQwen3.5 is a multi-modal ColBERT-style model based on Qwen3.5.
 It produces per-token embeddings and uses MaxSim scoring for retrieval
 and reranking. Supports both text and image inputs.

-Works for any ColQwen3.5 checkpoint, e.g. `athrael-soju/colqwen3.5-4.5B-v3`
-or `vultr/VultronRetrieverPrime-Qwen3.5-8B`.
-
 Start the server with:
-    vllm serve athrael-soju/colqwen3.5-4.5B-v3 --max-model-len 4096 \
-        --mm-processor-kwargs '{"min_pixels": 65536, "max_pixels": 1835008}'
+    vllm serve athrael-soju/colqwen3.5-4.5B --max-model-len 4096

 Then run this script:
    python colqwen3_5_rerank_online.py
-
-Parity note (matching the native colpali ColQwen3_5Processor pipeline):
-  - Visual-token budget: ColQwen3_5Processor uses max_num_visual_tokens=1792,
-    i.e. max_pixels = 1792 * (patch_size*merge_size)^2 = 1792 * 32^2 = 1835008
-    (with min_pixels = shortest_edge = 65536). Pass these via --mm-processor-kwargs
-    as above; the default budget gives fewer visual tokens and lower retrieval ndcg.
-  - When you build prompts yourself (token_embed), reproduce the processor exactly:
-      image (document): wrap in the instruction template
-        "<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>"
-        "Describe the image.<|im_end|><|endoftext|>"
-      query: append the augmentation suffix  <text> + "<|endoftext|>" * 10
-    Omitting these reproduces a silent ~2.5 ndcg@10 drop vs the native pipeline.
 """

 import requests
@@ -0,0 +1,91 @@
+{{ '<begin_of_document>' -}}
+{%- if custom_tools is defined %}
+    {%- set tools = custom_tools %}
+{%- endif %}
+{%- if not tools is defined %}
+    {%- set tools = none %}
+{%- endif %}
+
+{#- Extract system message #}
+{% set ns = namespace(system_prompt='') -%}
+{%- if messages[0]['role'] == 'system' %}
+    {%- if messages[0]['content'] is string %}
+        {%- set ns.system_prompt = messages[0]['content']|trim %}
+    {%- else %}
+        {%- set ns.system_prompt = messages[0]['content'][0]['text']|trim %}
+    {%- endif %}
+    {%- set messages = messages[1:] %}
+{%- else %}
+    {%- if tools is not none %}
+        {%- set ns.system_prompt = "You are a helpful assistant created by Minimax based on MiniMax-M1 model." %}
+    {%- else %}
+        {%- set ns.system_prompt = "You are a helpful assistant created by Minimax based on MiniMax-M1 model." %}
+    {%- endif %}
+{%- endif %}
+
+{#- System message #}
+{%- if ns.system_prompt != '' %}
+{{ '<beginning_of_sentence>system ai_setting=assistant\n' + ns.system_prompt + '<end_of_sentence>\n' -}}
+{%- endif %}
+
+{#- Tools configuration #}
+{%- if tools is not none %}
+{{ '<beginning_of_sentence>system tool_setting=tools\nYou are provided with these tools:\n<tools>\n' -}}
+{%- for tool in tools %}
+{{ tool | tojson ~ '\n' -}}
+{%- endfor %}
+{{ '</tools>\n\nIf you need to call tools, please respond with <tool_calls></tool_calls> XML tags, and provide tool-name and json-object of arguments, following the format below:\n<tool_calls>\n{"name": <tool-name>, "arguments": <args-json-object>}\n...\n</tool_calls><end_of_sentence>\n' -}}
+{%- endif %}
+
+{#- Process messages #}
+{%- for message in messages %}
+    {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
+        {%- if message['role'] == 'user' %}
+{{ '<beginning_of_sentence>user name=user\n' -}}
+{%- if message['content'] is string %}
+{{ message['content']|trim -}}
+{%- else %}
+{%- for content in message['content'] %}
+{%- if content['type'] == 'text' %}
+{{ content['text']|trim -}}
+{%- endif %}
+{%- endfor %}
+{%- endif %}
+{{ '<end_of_sentence>\n' -}}
+        {%- elif message['role'] == 'assistant' %}
+{{ '<beginning_of_sentence>ai name=assistant\n' -}}
+{%- if message['content'] is string %}
+{{ message['content']|trim -}}
+{%- else %}
+{%- for content in message['content'] | selectattr('type', 'equalto', 'text') %}
+{{ content['text']|trim -}}
+{%- endfor %}
+{%- endif %}
+{{ '<end_of_sentence>\n' -}}
+        {%- endif %}
+    {%- elif 'tool_calls' in message %}
+{{ '<beginning_of_sentence>ai name=assistant\n<tool_calls>\n' -}}
+{%- for tool_call in message.tool_calls %}
+{{ '{"name": "' + tool_call.function.name + '", "arguments": ' + tool_call.function.arguments | tojson + '}\n' -}}
+{%- endfor %}
+{{ '</tool_calls><end_of_sentence>\n' -}}
+    {%- elif message.role == "tool" or message.role == "ipython" %}
+{{ '<beginning_of_sentence>tool name=tools\n' -}}
+{%- if message.content is string %}
+{{ 'tool result: ' + message.content + '\n\n' -}}
+{%- else %}
+{%- for content in message['content'] %}
+{%- if content['type'] == 'text' %}
+{{ 'tool result: ' + content['text'] + '\n\n' -}}
+{%- elif content.get('name') %}
+{{ 'tool name: ' + content['name'] + '\ntool result: ' + content['text'] + '\n\n' -}}
+{%- endif %}
+{%- endfor %}
+{%- endif %}
+{{ '<end_of_sentence>\n' -}}
+    {%- endif %}
+{%- endfor %}
+
+{%- if add_generation_prompt %}
+{{ '<beginning_of_sentence>ai name=assistant\n' -}}
+{%- endif %}
@@ -11,14 +11,13 @@ transformers >= 5.5.3
 tokenizers >= 0.21.1  # Required for fast incremental detokenization.
 safetensors >= 0.6.2  # MXFP4/MXFP6 dtype support (F8_E8M0, F4) added in 0.6.0: https://github.com/huggingface/safetensors/pull/611
 protobuf >= 5.29.6, !=6.30.*, !=6.31.*, !=6.32.*, !=6.33.0.*, !=6.33.1.*, !=6.33.2.*, !=6.33.3.*, !=6.33.4.* # Required by LlamaTokenizer, gRPC. CVE-2026-0994
-fastapi[standard] >= 0.133.0, < 0.137.0 # First version supporting Starlette 1.0; < 0.137.0 avoids route-tree change that breaks model-hosting-container-standards handler overrides.
-starlette >= 1.0.1 # CVE-2026-48710: Host header injection in < 1.0.1
+fastapi[standard] >= 0.115.0 # Required by FastAPI's form models in the OpenAI API server's audio transcriptions endpoint.
 aiohttp >= 3.13.3
 openai >= 2.0.0  # For Responses API with reasoning content
 pydantic >= 2.12.0
 prometheus_client >= 0.18.0
 pillow  # Required for image processing
-prometheus-fastapi-instrumentator >= 8.0.0 # v8 unblocks starlette >= 1.0
+prometheus-fastapi-instrumentator >= 7.0.0
 tiktoken >= 0.6.0  # Required for DBRX tokenizer
 lm-format-enforcer == 0.11.3
 llguidance >= 1.7.0, < 1.8.0; platform_machine == "x86_64" or platform_machine == "arm64" or platform_machine == "aarch64" or platform_machine == "ppc64le"
@@ -11,7 +11,7 @@ numba == 0.65.0 # Required for N-gram speculative decoding
 datasets
 peft
 pytest-asyncio
-tensorizer==2.10.1
+tensorizer==2.12.1
 packaging>=24.2
 setuptools>=77.0.3,<80.0.0
 setuptools-scm>=8
@@ -1,6 +1,6 @@
 # testing
 pytest
-tensorizer==2.10.1
+tensorizer==2.12.1
 pytest-forked
 pytest-asyncio
 pytest-rerunfailures
@@ -40,7 +40,7 @@ lm-eval[api]>=0.4.12 # required for model evaluation test
 mteb[bm25s]>=2, <3 # required for mteb test
 transformers==5.5.3
 tokenizers==0.22.2
-schemathesis>=4.0.0 # Required for openai schema test.
+schemathesis>=3.39.15 # Required for openai schema test.
 # quantization
 bitsandbytes==0.49.2
 buildkite-test-collector==0.1.9
@@ -35,11 +35,14 @@ arctic-inference==0.1.1
    # via -r requirements/test/cuda.in
 argcomplete==3.5.1
    # via datamodel-code-generator
+arrow==1.3.0
+    # via isoduration
 attrs==24.2.0
    # via
    #   aiohttp
    #   hypothesis
    #   jsonschema
+    #   pytest-subtests
    #   referencing
 audioread==3.0.1
    # via librosa
@@ -54,7 +57,9 @@ azure-identity==1.25.2
 azure-storage-blob==12.28.0
    # via runai-model-streamer-azure
 backoff==2.2.1
-    # via -r requirements/test/cuda.in
+    # via
+    #   -r requirements/test/cuda.in
+    #   schemathesis
 bitsandbytes==0.49.2
    # via -r requirements/test/cuda.in
 black==24.10.0
@@ -105,6 +110,7 @@ colorama==0.4.6
    # via
    #   perceptron
    #   sacrebleu
+    #   schemathesis
 colorful==0.5.6
    # via ray
 colorlog==6.10.1
@@ -177,7 +183,7 @@ et-xmlfile==2.0.0
    # via openpyxl
 evaluate==0.4.3
    # via lm-eval
-fastapi==0.136.3
+fastapi==0.128.0
    # via
    #   -c requirements/common.txt
    #   gpt-oss
@@ -200,6 +206,8 @@ filelock==3.16.1
    #   virtualenv
 fonttools==4.55.0
    # via matplotlib
+fqdn==1.5.1
+    # via jsonschema
 frozendict==2.4.6
    # via einx
 frozenlist==1.5.0
@@ -261,7 +269,7 @@ h11==0.14.0
    #   uvicorn
 h2==4.3.0
    # via httpx
-harfile==0.5.0
+harfile==0.3.0
    # via schemathesis
 hf-xet==1.4.3
    # via huggingface-hub
@@ -301,7 +309,7 @@ hypothesis==6.131.0
    #   hypothesis-graphql
    #   hypothesis-jsonschema
    #   schemathesis
-hypothesis-graphql==0.13.0
+hypothesis-graphql==0.11.1
    # via schemathesis
 hypothesis-jsonschema==0.23.1
    # via schemathesis
@@ -310,6 +318,7 @@ idna==3.10
    #   anyio
    #   email-validator
    #   httpx
+    #   jsonschema
    #   requests
    #   yarl
 imagehash==4.3.2
@@ -326,6 +335,8 @@ instanttensor==0.1.5
    # via -r requirements/test/cuda.in
 isodate==0.7.2
    # via azure-storage-blob
+isoduration==20.11.0
+    # via jsonschema
 isort==5.13.2
    # via datamodel-code-generator
 jinja2==3.1.6
@@ -345,14 +356,15 @@ joblib==1.4.2
    #   librosa
    #   nltk
    #   scikit-learn
+jsonpointer==3.0.0
+    # via jsonschema
 jsonschema==4.23.0
    # via
    #   -c requirements/common.txt
    #   hypothesis-jsonschema
    #   mistral-common
    #   ray
-jsonschema-rs==0.46.5
-    # via schemathesis
+    #   schemathesis
 jsonschema-specifications==2024.10.1
    # via jsonschema
 junit-xml==1.9
@@ -703,20 +715,18 @@ pydantic-core==2.41.1
 pydantic-extra-types==2.10.5
    # via mistral-common
 pygments==2.18.0
-    # via
-    #   pytest
-    #   rich
+    # via rich
 pyjwt==2.11.0
    # via msal
 pyparsing==3.2.0
    # via matplotlib
-pyrate-limiter==4.4.0
+pyrate-limiter==3.7.0
    # via schemathesis
 pystemmer==3.0.0
    # via mteb
 pytablewriter==1.2.0
    # via lm-eval
-pytest==9.1.0
+pytest==8.3.5
    # via
    #   -r requirements/test/cuda.in
    #   buildkite-test-collector
@@ -727,9 +737,10 @@ pytest==9.1.0
    #   pytest-mock
    #   pytest-rerunfailures
    #   pytest-shard
+    #   pytest-subtests
    #   pytest-timeout
    #   schemathesis
-pytest-asyncio==1.4.0
+pytest-asyncio==0.24.0
    # via -r requirements/test/cuda.in
 pytest-cov==6.3.0
    # via -r requirements/test/cuda.in
@@ -741,10 +752,13 @@ pytest-rerunfailures==14.0
    # via -r requirements/test/cuda.in
 pytest-shard==0.1.2
    # via -r requirements/test/cuda.in
+pytest-subtests==0.14.1
+    # via schemathesis
 pytest-timeout==2.3.1
    # via -r requirements/test/cuda.in
 python-dateutil==2.9.0.post0
    # via
+    #   arrow
    #   botocore
    #   matplotlib
    #   pandas
@@ -815,12 +829,15 @@ requests==2.32.3
    #   tiktoken
 responses==0.25.3
    # via genai-perf
+rfc3339-validator==0.1.4
+    # via jsonschema
+rfc3987==1.3.8
+    # via jsonschema
 rich==13.9.4
    # via
    #   genai-perf
    #   mteb
    #   perceptron
-    #   schemathesis
    #   typer
 rouge-score==0.1.2
    # via lm-eval
@@ -851,7 +868,7 @@ safetensors==0.7.0
    #   segmentation-models-pytorch
    #   timm
    #   transformers
-schemathesis==4.21.6
+schemathesis==3.39.15
    # via -r requirements/test/cuda.in
 scikit-image==0.25.2
    # via albumentations
@@ -895,6 +912,7 @@ six==1.16.0
    #   junit-xml
    #   opencensus
    #   python-dateutil
+    #   rfc3339-validator
    #   rouge-score
 smart-open==7.1.0
    # via ray
@@ -920,10 +938,10 @@ sqlalchemy==2.0.41
    #   optuna
 sqlitedict==2.1.0
    # via lm-eval
-starlette==1.3.1
+starlette==0.50.0
    # via
-    #   -c requirements/common.txt
    #   fastapi
+    #   schemathesis
    #   starlette-testclient
 starlette-testclient==0.4.1
    # via schemathesis
@@ -948,8 +966,7 @@ tenacity==9.1.2
    #   gpt-oss
    #   lm-eval
    #   plotly
-    #   schemathesis
-tensorizer==2.10.1
+tensorizer==2.12.1
    # via -r requirements/test/cuda.in
 termcolor==3.1.0
    # via gpt-oss
@@ -973,6 +990,10 @@ tokenizers==0.22.2
    #   -c requirements/common.txt
    #   -r requirements/test/cuda.in
    #   transformers
+tomli==2.2.1
+    # via schemathesis
+tomli-w==1.2.0
+    # via schemathesis
 torch==2.11.0+cu130
    # via
    #   -c requirements/cuda.txt
@@ -1045,6 +1066,8 @@ typer==0.15.2
    #   huggingface-hub
    #   perceptron
    #   transformers
+types-python-dateutil==2.9.0.20241206
+    # via arrow
 typing-extensions==4.15.0
    # via
    #   -c requirements/common.txt
@@ -1069,8 +1092,6 @@ typing-extensions==4.15.0
    #   pydantic
    #   pydantic-core
    #   pydantic-extra-types
-    #   pytest-asyncio
-    #   schemathesis
    #   sentence-transformers
    #   sqlalchemy
    #   starlette
@@ -1078,11 +1099,11 @@ typing-extensions==4.15.0
    #   typer
    #   typing-inspection
 typing-inspection==0.4.2
-    # via
-    #   fastapi
-    #   pydantic
+    # via pydantic
 tzdata==2024.2
    # via pandas
+uri-template==1.3.0
+    # via jsonschema
 urllib3==2.2.3
    # via
    #   blobfile
@@ -1101,6 +1122,8 @@ vocos==0.1.0
    # via -r requirements/test/cuda.in
 wcwidth==0.2.13
    # via ftfy
+webcolors==24.11.1
+    # via jsonschema
 werkzeug==3.1.3
    # via schemathesis
 word2number==1.1
@@ -1112,6 +1135,8 @@ xxhash==3.5.0
    #   datasets
    #   evaluate
 yarl==1.17.1
-    # via aiohttp
+    # via
+    #   aiohttp
+    #   schemathesis
 zipp==3.23.0
    # via importlib-metadata
@@ -1,6 +1,6 @@
 # testing
 pytest
-tensorizer==2.10.1
+tensorizer==2.12.1
 pytest-forked
 pytest-asyncio
 pytest-rerunfailures
@@ -31,7 +31,7 @@ lm-eval[api]>=0.4.12 # required for model evaluation test
 mteb[bm25s]>=2, <3 # required for mteb test
 transformers==5.5.3
 tokenizers==0.22.2
-schemathesis>=4.0.0 # Required for openai schema test.
+schemathesis>=3.39.15 # Required for openai schema test.
 # quantization
 bitsandbytes>=0.49.2
 buildkite-test-collector==0.1.9
@@ -2,7 +2,7 @@

 # testing
 pytest
-tensorizer==2.10.1
+tensorizer==2.12.1
 pytest-forked
 pytest-asyncio
 pytest-rerunfailures
@@ -39,7 +39,7 @@ lm-eval[api]>=0.4.12 # required for model evaluation test
 mteb[bm25s]>=2, <3 # required for mteb test
 transformers==5.5.3
 tokenizers==0.22.2
-schemathesis>=4.0.0 # Required for openai schema test
+schemathesis>=3.39.15 # Required for openai schema test
 # quantization
 bitsandbytes==0.49.2
 buildkite-test-collector==0.1.9
@@ -51,12 +51,15 @@ arctic-inference==0.1.1
    # via -r requirements/test/rocm.in
 argcomplete==3.6.3
    # via datamodel-code-generator
+arrow==1.4.0
+    # via isoduration
 astor==0.8.1
    # via depyf
 attrs==26.1.0
    # via
    #   aiohttp
    #   jsonschema
+    #   pytest-subtests
    #   referencing
 audioread==3.0.1
    # via librosa
@@ -71,7 +74,9 @@ azure-identity==1.25.3
 azure-storage-blob==12.28.0
    # via runai-model-streamer-azure
 backoff==2.2.1
-    # via -r requirements/test/rocm.in
+    # via
+    #   -r requirements/test/rocm.in
+    #   schemathesis
 bitsandbytes==0.49.2
    # via -r requirements/test/rocm.in
 black==26.3.1
@@ -134,6 +139,7 @@ colorama==0.4.6
    # via
    #   perceptron
    #   sacrebleu
+    #   schemathesis
 colorful==0.5.8
    # via ray
 colorlog==6.10.1
@@ -252,6 +258,8 @@ filelock==3.25.2
    #   virtualenv
 fonttools==4.62.1
    # via matplotlib
+fqdn==1.5.1
+    # via jsonschema
 frozendict==2.4.7
    # via einx
 frozenlist==1.8.0
@@ -320,7 +328,7 @@ h11==0.16.0
    #   uvicorn
 h2==4.3.0
    # via httpx
-harfile==0.5.0
+harfile==0.4.0
    # via schemathesis
 hf-xet==1.4.3
    # via huggingface-hub
@@ -370,7 +378,7 @@ hypothesis==6.151.9
    #   hypothesis-graphql
    #   hypothesis-jsonschema
    #   schemathesis
-hypothesis-graphql==0.13.0
+hypothesis-graphql==0.12.0
    # via schemathesis
 hypothesis-jsonschema==0.23.1
    # via schemathesis
@@ -379,6 +387,7 @@ idna==3.11
    #   anyio
    #   email-validator
    #   httpx
+    #   jsonschema
    #   requests
    #   yarl
 ijson==3.5.0
@@ -399,6 +408,8 @@ interegular==0.3.3
    # via lm-format-enforcer
 isodate==0.7.2
    # via azure-storage-blob
+isoduration==20.11.0
+    # via jsonschema
 isort==8.0.1
    # via datamodel-code-generator
 jinja2==3.1.6
@@ -424,6 +435,8 @@ joblib==1.5.3
    #   librosa
    #   nltk
    #   scikit-learn
+jsonpointer==3.1.0
+    # via jsonschema
 jsonschema==4.26.0
    # via
    #   -c requirements/common.txt
@@ -432,8 +445,7 @@ jsonschema==4.26.0
    #   mcp
    #   mistral-common
    #   ray
-jsonschema-rs==0.46.5
-    # via schemathesis
+    #   schemathesis
 jsonschema-specifications==2025.9.1
    # via jsonschema
 junit-xml==1.9
@@ -780,7 +792,7 @@ prometheus-client==0.24.1
    #   opentelemetry-exporter-prometheus
    #   prometheus-fastapi-instrumentator
    #   ray
-prometheus-fastapi-instrumentator==8.0.0
+prometheus-fastapi-instrumentator==7.1.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
@@ -864,22 +876,20 @@ pydantic-settings==2.13.1
    #   fastapi
    #   mcp
 pygments==2.19.2
-    # via
-    #   pytest
-    #   rich
+    # via rich
 pyjwt==2.12.1
    # via
    #   mcp
    #   msal
 pyparsing==3.3.2
    # via matplotlib
-pyrate-limiter==4.4.0
+pyrate-limiter==3.9.0
    # via schemathesis
 pystemmer==3.0.0
    # via mteb
 pytablewriter==1.2.1
    # via lm-eval
-pytest==9.1.0
+pytest==8.3.5
    # via
    #   -r requirements/test/rocm.in
    #   buildkite-test-collector
@@ -890,9 +900,10 @@ pytest==9.1.0
    #   pytest-mock
    #   pytest-rerunfailures
    #   pytest-shard
+    #   pytest-subtests
    #   pytest-timeout
    #   schemathesis
-pytest-asyncio==1.4.0
+pytest-asyncio==0.24.0
    # via -r requirements/test/rocm.in
 pytest-cov==6.3.0
    # via -r requirements/test/rocm.in
@@ -904,10 +915,13 @@ pytest-rerunfailures==14.0
    # via -r requirements/test/rocm.in
 pytest-shard==0.1.2
    # via -r requirements/test/rocm.in
+pytest-subtests==0.14.2
+    # via schemathesis
 pytest-timeout==2.3.1
    # via -r requirements/test/rocm.in
 python-dateutil==2.9.0.post0
    # via
+    #   arrow
    #   botocore
    #   matplotlib
    #   pandas
@@ -1002,13 +1016,16 @@ requests==2.32.5
    #   tiktoken
 responses==0.26.0
    # via genai-perf
+rfc3339-validator==0.1.4
+    # via jsonschema
+rfc3987==1.3.8
+    # via jsonschema
 rich==14.3.3
    # via
    #   genai-perf
    #   mteb
    #   perceptron
    #   rich-toolkit
-    #   schemathesis
    #   typer
 rich-toolkit==0.19.7
    # via
@@ -1046,7 +1063,7 @@ safetensors==0.7.0
    #   segmentation-models-pytorch
    #   timm
    #   transformers
-schemathesis==4.21.6
+schemathesis==3.39.15
    # via -r requirements/test/rocm.in
 scikit-image==0.26.0
    # via albumentations
@@ -1103,6 +1120,7 @@ six==1.17.0
    #   junit-xml
    #   opencensus
    #   python-dateutil
+    #   rfc3339-validator
    #   rouge-score
 smart-open==7.5.1
    # via ray
@@ -1131,14 +1149,13 @@ sqlitedict==2.1.0
    # via lm-eval
 sse-starlette==3.3.4
    # via mcp
-starlette==1.3.1
+starlette==0.52.1
    # via
-    #   -c requirements/common.txt
-    #   -r requirements/test/../common.txt
    #   fastapi
    #   mcp
    #   model-hosting-container-standards
    #   prometheus-fastapi-instrumentator
+    #   schemathesis
    #   sse-starlette
    #   starlette-testclient
 starlette-testclient==0.4.1
@@ -1165,8 +1182,7 @@ tenacity==9.1.4
    # via
    #   gpt-oss
    #   lm-eval
-    #   schemathesis
-tensorizer==2.10.1
+tensorizer==2.12.1
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
@@ -1199,6 +1215,10 @@ tokenizers==0.22.2
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   transformers
+tomli==2.4.0
+    # via schemathesis
+tomli-w==1.2.0
+    # via schemathesis
 torch-c-dlpack-ext==0.1.5
    # via tilelang
 tqdm==4.67.3
@@ -1281,10 +1301,8 @@ typing-extensions==4.15.0
    #   pydantic
    #   pydantic-core
    #   pydantic-extra-types
-    #   pytest-asyncio
    #   referencing
    #   rich-toolkit
-    #   schemathesis
    #   sentence-transformers
    #   sqlalchemy
    #   starlette
@@ -1299,6 +1317,10 @@ typing-inspection==0.4.2
    #   mcp
    #   pydantic
    #   pydantic-settings
+tzdata==2025.3
+    # via arrow
+uri-template==1.3.0
+    # via jsonschema
 urllib3==2.6.3
    # via
    #   blobfile
@@ -1329,6 +1351,8 @@ watchfiles==1.1.1
    #   uvicorn
 wcwidth==0.6.0
    # via ftfy
+webcolors==25.10.0
+    # via jsonschema
 websockets==16.0
    # via uvicorn
 werkzeug==3.1.6
@@ -1346,7 +1370,9 @@ xxhash==3.6.0
    #   datasets
    #   evaluate
 yarl==1.23.0
-    # via aiohttp
+    # via
+    #   aiohttp
+    #   schemathesis
 z3-solver==4.15.4.0
    # via tilelang
 zipp==3.23.0
@@ -593,9 +593,8 @@ soxr==0.5.0.post1
    #   mistral-common
 sqlitedict==2.1.0
    # via lm-eval
-starlette==1.3.1
+starlette==1.0.0
    # via
-    #   -c requirements/common.txt
    #   fastapi
    #   starlette-testclient
 starlette-testclient==0.4.1
@@ -17,4 +17,4 @@ torchaudio
 torchvision

 auto_round_lib>=0.13.3
-vllm_xpu_kernels @ https://github.com/vllm-project/vllm-xpu-kernels/releases/download/v0.1.10/vllm_xpu_kernels-0.1.10-cp38-abi3-manylinux_2_28_x86_64.whl
+vllm_xpu_kernels @ https://github.com/vllm-project/vllm-xpu-kernels/releases/download/v0.1.9.1/vllm_xpu_kernels-0.1.9.1-cp38-abi3-manylinux_2_28_x86_64.whl
@@ -386,6 +386,7 @@ mod tests {
            tool_chat_template_llama3.2_pythonic.jinja         => String
            tool_chat_template_llama4_json.jinja               => OpenAi
            tool_chat_template_llama4_pythonic.jinja           => OpenAi
+            tool_chat_template_minimax_m1.jinja                => OpenAi
            tool_chat_template_mistral.jinja                   => String
            tool_chat_template_mistral3.jinja                  => OpenAi
            tool_chat_template_mistral_parallel.jinja          => String
@@ -16,8 +16,7 @@ use vllm_engine_core_client::protocol::logprobs::{
    Logprobs, MaybeWireLogprobs, PositionLogprobs, TokenLogprob,
 };
 use vllm_engine_core_client::protocol::{
-    EngineCoreFinishReason, EngineCoreOutput, EngineCoreOutputs, EngineCoreRequest, LogprobsCount,
-    StopReason,
+    EngineCoreFinishReason, EngineCoreOutput, EngineCoreOutputs, EngineCoreRequest, StopReason,
 };
 use vllm_engine_core_client::test_utils::{IpcNamespace, spawn_mock_engine_task};
 use vllm_engine_core_client::{EngineCoreClient, EngineCoreClientConfig};
@@ -1388,8 +1387,8 @@ async fn chat_stream_and_collect_preserve_prompt_and_sample_logprobs() {
    .await;

    let mut request = sample_request("chat-logprobs");
-    request.sampling_params.logprobs = Some(LogprobsCount::Top(1));
-    request.sampling_params.prompt_logprobs = Some(LogprobsCount::Top(1));
+    request.sampling_params.logprobs = Some(1);
+    request.sampling_params.prompt_logprobs = Some(1);

    let mut stream = chat.chat(request.clone()).await.unwrap();
    match next_semantic(&mut stream).await.unwrap().unwrap() {
@@ -0,0 +1,91 @@
+{{ '<begin_of_document>' -}}
+{%- if custom_tools is defined %}
+    {%- set tools = custom_tools %}
+{%- endif %}
+{%- if not tools is defined %}
+    {%- set tools = none %}
+{%- endif %}
+
+{#- Extract system message #}
+{% set ns = namespace(system_prompt='') -%}
+{%- if messages[0]['role'] == 'system' %}
+    {%- if messages[0]['content'] is string %}
+        {%- set ns.system_prompt = messages[0]['content']|trim %}
+    {%- else %}
+        {%- set ns.system_prompt = messages[0]['content'][0]['text']|trim %}
+    {%- endif %}
+    {%- set messages = messages[1:] %}
+{%- else %}
+    {%- if tools is not none %}
+        {%- set ns.system_prompt = "You are a helpful assistant created by Minimax based on MiniMax-M1 model." %}
+    {%- else %}
+        {%- set ns.system_prompt = "You are a helpful assistant created by Minimax based on MiniMax-M1 model." %}
+    {%- endif %}
+{%- endif %}
+
+{#- System message #}
+{%- if ns.system_prompt != '' %}
+{{ '<beginning_of_sentence>system ai_setting=assistant\n' + ns.system_prompt + '<end_of_sentence>\n' -}}
+{%- endif %}
+
+{#- Tools configuration #}
+{%- if tools is not none %}
+{{ '<beginning_of_sentence>system tool_setting=tools\nYou are provided with these tools:\n<tools>\n' -}}
+{%- for tool in tools %}
+{{ tool | tojson ~ '\n' -}}
+{%- endfor %}
+{{ '</tools>\n\nIf you need to call tools, please respond with <tool_calls></tool_calls> XML tags, and provide tool-name and json-object of arguments, following the format below:\n<tool_calls>\n{"name": <tool-name>, "arguments": <args-json-object>}\n...\n</tool_calls><end_of_sentence>\n' -}}
+{%- endif %}
+
+{#- Process messages #}
+{%- for message in messages %}
+    {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}
+        {%- if message['role'] == 'user' %}
+{{ '<beginning_of_sentence>user name=user\n' -}}
+{%- if message['content'] is string %}
+{{ message['content']|trim -}}
+{%- else %}
+{%- for content in message['content'] %}
+{%- if content['type'] == 'text' %}
+{{ content['text']|trim -}}
+{%- endif %}
+{%- endfor %}
+{%- endif %}
+{{ '<end_of_sentence>\n' -}}
+        {%- elif message['role'] == 'assistant' %}
+{{ '<beginning_of_sentence>ai name=assistant\n' -}}
+{%- if message['content'] is string %}
+{{ message['content']|trim -}}
+{%- else %}
+{%- for content in message['content'] | selectattr('type', 'equalto', 'text') %}
+{{ content['text']|trim -}}
+{%- endfor %}
+{%- endif %}
+{{ '<end_of_sentence>\n' -}}
+        {%- endif %}
+    {%- elif 'tool_calls' in message %}
+{{ '<beginning_of_sentence>ai name=assistant\n<tool_calls>\n' -}}
+{%- for tool_call in message.tool_calls %}
+{{ '{"name": "' + tool_call.function.name + '", "arguments": ' + tool_call.function.arguments | tojson + '}\n' -}}
+{%- endfor %}
+{{ '</tool_calls><end_of_sentence>\n' -}}
+    {%- elif message.role == "tool" or message.role == "ipython" %}
+{{ '<beginning_of_sentence>tool name=tools\n' -}}
+{%- if message.content is string %}
+{{ 'tool result: ' + message.content + '\n\n' -}}
+{%- else %}
+{%- for content in message['content'] %}
+{%- if content['type'] == 'text' %}
+{{ 'tool result: ' + content['text'] + '\n\n' -}}
+{%- elif content.get('name') %}
+{{ 'tool name: ' + content['name'] + '\ntool result: ' + content['text'] + '\n\n' -}}
+{%- endif %}
+{%- endfor %}
+{%- endif %}
+{{ '<end_of_sentence>\n' -}}
+    {%- endif %}
+{%- endfor %}
+
+{%- if add_generation_prompt %}
+{{ '<beginning_of_sentence>ai name=assistant\n' -}}
+{%- endif %}
@@ -20,7 +20,6 @@ use serde_with::{DefaultOnNull, OneOrMany, serde_as};
 use thiserror_ext::AsReport as _;
 use uuid::Uuid;
 use vllm_engine_core_client::TransportMode;
-use vllm_engine_core_client::protocol::LogprobsCount;
 use vllm_managed_engine::ManagedEngineConfig;
 use vllm_managed_engine::cli::{ManagedEngineArgs, repartition_managed_engine_args};
 use vllm_server::{
@@ -137,9 +136,9 @@ pub struct SharedRuntimeArgs {
    pub max_model_len: Option<u32>,
    /// Maximum number of log probabilities to return when `logprobs` is
    /// specified in sampling parameters. `-1` means no cap.
-    #[arg(long, allow_negative_numbers = true)]
+    #[arg(long, value_parser = clap::value_parser!(i32).range(-1..), allow_negative_numbers = true)]
    #[serde(default)]
-    pub max_logprobs: Option<LogprobsCount>,
+    pub max_logprobs: Option<i32>,
    /// TCP port for the gRPC Generate service. When not set, no gRPC server is
    /// started.
    #[arg(long)]
@@ -530,7 +529,7 @@ impl ServeArgs {
        self.managed_engine.clone().into_config(
            self.runtime.model.clone(),
            self.runtime.max_model_len,
-            self.runtime.max_logprobs.map(managed_max_logprobs_to_i32),
+            self.runtime.max_logprobs,
            self.runtime.language_model_only,
            self.runtime.disable_log_stats,
            self.runtime.shutdown_timeout,
@@ -556,9 +555,5 @@ fn frontend_ipc_addresses() -> (String, String) {
    )
 }

-fn managed_max_logprobs_to_i32(count: LogprobsCount) -> i32 {
-    i32::try_from(count).expect("max_logprobs is parsed through i32")
-}
-
 #[cfg(test)]
 mod tests;
@@ -1,6 +1,5 @@
 use expect_test::expect;
 use vllm_engine_core_client::TransportMode;
-use vllm_engine_core_client::protocol::LogprobsCount;
 use vllm_server::{Config, HttpListenerMode, ParserSelection, RendererSelection};

 use super::{Cli, Command};
@@ -166,10 +165,10 @@ fn serve_args_forward_max_logprobs_to_frontend_and_managed_engine() {
    let Command::Serve(args) = cli.command else {
        panic!("expected serve args");
    };
-    assert_eq!(args.runtime.max_logprobs, Some(LogprobsCount::All));
+    assert_eq!(args.runtime.max_logprobs, Some(-1));

    let frontend_config = args.to_frontend_config("tcp://127.0.0.1:62100".to_string());
-    assert_eq!(frontend_config.max_logprobs, Some(LogprobsCount::All));
+    assert_eq!(frontend_config.max_logprobs, Some(-1));

    let engine_config = args.to_managed_engine_config(5555);
    assert_eq!(engine_config.python_args, vec!["--max-logprobs", "-1"]);
@@ -530,7 +529,7 @@ fn frontend_args_json_accepts_supported_non_default_fields() {
    assert_eq!(args.runtime.renderer, RendererSelection::DeepSeekV32);
    assert!(args.runtime.language_model_only);
    assert_eq!(args.runtime.max_model_len, Some(8192));
-    assert_eq!(args.runtime.max_logprobs, Some(LogprobsCount::All));
+    assert_eq!(args.runtime.max_logprobs, Some(-1));
    assert_eq!(args.runtime.shutdown_timeout, 3);
 }

@@ -6,7 +6,7 @@ use futures::StreamExt as _;
 use tokio::time::timeout;
 use tracing_subscriber::EnvFilter;
 use vllm_engine_core_client::protocol::{
-    EngineCoreFinishReason, EngineCoreRequest, EngineCoreSamplingParams, LogprobsCount,
+    EngineCoreFinishReason, EngineCoreRequest, EngineCoreSamplingParams,
 };
 use vllm_engine_core_client::{
    EngineCoreClient, EngineCoreClientConfig, EngineCoreStreamOutput, TransportMode,
@@ -33,10 +33,10 @@ struct Args {
    output_timeout_secs: u64,
    #[arg(long, default_value_t = 1)]
    max_tokens: u32,
-    #[arg(long, default_value_t = LogprobsCount::Top(2), allow_negative_numbers = true)]
-    logprobs: LogprobsCount,
-    #[arg(long, default_value_t = LogprobsCount::Top(1), allow_negative_numbers = true)]
-    prompt_logprobs: LogprobsCount,
+    #[arg(long, default_value_t = 2)]
+    logprobs: i32,
+    #[arg(long, default_value_t = 1)]
+    prompt_logprobs: i32,
    #[arg(long, default_value_t = 96)]
    prompt_repeats: usize,
 }
@@ -64,8 +64,8 @@ fn build_request(
    request_id: String,
    prompt_token_ids: Vec<u32>,
    max_tokens: u32,
-    logprobs: LogprobsCount,
-    prompt_logprobs: LogprobsCount,
+    logprobs: i32,
+    prompt_logprobs: i32,
    client_index: u32,
 ) -> EngineCoreRequest {
    EngineCoreRequest {
@@ -1,135 +0,0 @@
-use std::fmt;
-use std::str::FromStr;
-
-use serde::{Deserialize, Deserializer, Serialize, Serializer};
-
-/// Number of log probabilities requested for a token position.
-#[derive(Debug, Clone, Copy, PartialEq, Eq)]
-pub enum LogprobsCount {
-    /// Return the full model vocabulary.
-    All,
-    /// Return the top-N tokens by probability.
-    Top(u32),
-}
-
-impl LogprobsCount {
-    /// Expands the count to the actual number of logprobs to return, given the vocabulary size.
-    pub fn expanded(self, vocab_size: usize) -> usize {
-        match self {
-            Self::All => vocab_size,
-            Self::Top(count) => count as usize,
-        }
-    }
-}
-
-impl TryFrom<i32> for LogprobsCount {
-    type Error = String;
-
-    fn try_from(value: i32) -> Result<Self, Self::Error> {
-        match value {
-            -1 => Ok(Self::All),
-            value if value < -1 => Err(format!("must be non-negative or -1, got {value}")),
-            value => Ok(Self::Top(value as u32)),
-        }
-    }
-}
-
-impl TryFrom<LogprobsCount> for i32 {
-    type Error = String;
-
-    fn try_from(value: LogprobsCount) -> Result<Self, Self::Error> {
-        match value {
-            LogprobsCount::All => Ok(-1),
-            LogprobsCount::Top(count) => {
-                i32::try_from(count).map_err(|_| format!("must fit within i32, got {count}"))
-            }
-        }
-    }
-}
-
-impl FromStr for LogprobsCount {
-    type Err = String;
-
-    fn from_str(s: &str) -> Result<Self, Self::Err> {
-        let value = s
-            .parse::<i32>()
-            .map_err(|e| format!("must be an i32 integer, got {s:?}: {e}"))?;
-        Self::try_from(value)
-    }
-}
-
-impl fmt::Display for LogprobsCount {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        match self {
-            Self::All => (-1).fmt(f),
-            Self::Top(count) => count.fmt(f),
-        }
-    }
-}
-
-impl Serialize for LogprobsCount {
-    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
-    where
-        S: Serializer,
-    {
-        let value: i32 = (*self).try_into().map_err(serde::ser::Error::custom)?;
-        value.serialize(serializer)
-    }
-}
-
-impl<'de> Deserialize<'de> for LogprobsCount {
-    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
-    where
-        D: Deserializer<'de>,
-    {
-        let value = i32::deserialize(deserializer)?;
-        Self::try_from(value).map_err(serde::de::Error::custom)
-    }
-}
-
-#[cfg(test)]
-mod tests {
-    use rmpv::Value;
-
-    use super::*;
-    use crate::protocol::{decode_msgpack, encode_msgpack};
-
-    #[test]
-    fn logprobs_count_serializes_as_wire_integer() {
-        assert_eq!(serde_json::to_value(LogprobsCount::All).unwrap(), -1);
-        assert_eq!(serde_json::to_value(LogprobsCount::Top(3)).unwrap(), 3);
-    }
-
-    #[test]
-    fn logprobs_count_deserializes_wire_integer() {
-        assert_eq!(
-            serde_json::from_value::<LogprobsCount>(serde_json::json!(-1)).unwrap(),
-            LogprobsCount::All
-        );
-        assert_eq!(
-            serde_json::from_value::<LogprobsCount>(serde_json::json!(3)).unwrap(),
-            LogprobsCount::Top(3)
-        );
-        assert!(serde_json::from_value::<LogprobsCount>(serde_json::json!(-2)).is_err());
-        assert!(
-            serde_json::from_value::<LogprobsCount>(serde_json::json!(i64::from(i32::MAX) + 1))
-                .is_err()
-        );
-    }
-
-    #[test]
-    fn logprobs_count_decodes_msgpack_signed_and_unsigned() {
-        let mut encoded = Vec::new();
-        rmpv::encode::write_value(&mut encoded, &Value::from(-1)).unwrap();
-        assert_eq!(
-            decode_msgpack::<LogprobsCount>(&encoded).unwrap(),
-            LogprobsCount::All
-        );
-
-        let encoded = encode_msgpack(&LogprobsCount::Top(7)).unwrap();
-        assert_eq!(
-            decode_msgpack::<LogprobsCount>(&encoded).unwrap(),
-            LogprobsCount::Top(7)
-        );
-    }
-}
@@ -56,7 +56,6 @@ mod classified_outputs;
 pub mod dtype;
 pub mod handshake;
 pub mod logprobs;
-mod logprobs_count;
 pub mod lora;
 pub mod multimodal;
 pub mod stats;
@@ -67,7 +66,6 @@ pub use classified_outputs::{
 };
 pub use dtype::ModelDtype;
 pub use logprobs::decode_engine_core_outputs;
-pub use logprobs_count::LogprobsCount;

 /// Request types are encoded as single-byte protocol constants so they can be
 /// sent over the ZMQ socket without an extra encoding step.
@@ -279,20 +277,14 @@ pub struct EngineCoreSamplingParams {
    pub max_tokens: u32,
    /// Minimum number of tokens to generate before EOS or stop-token handling.
    pub min_tokens: u32,
-    /// Maximum number of reasoning ("thinking") tokens to emit before the
-    /// reasoning section is force-closed. `None` means unlimited; the
-    /// user-facing `-1` sentinel is normalized to `None` by the frontend before
-    /// reaching this DTO, so only non-negative values are sent. Enforced
-    /// engine-side (and only when a reasoning parser is configured).
-    pub thinking_token_budget: Option<u64>,
    /// Number of log probabilities to return per generated token.
    ///
-    /// `None` disables sample logprobs.
-    pub logprobs: Option<LogprobsCount>,
+    /// `None` disables sample logprobs. `-1` requests the full vocabulary.
+    pub logprobs: Option<i32>,
    /// Number of log probabilities to return per prompt token.
    ///
-    /// `None` disables prompt logprobs.
-    pub prompt_logprobs: Option<LogprobsCount>,
+    /// `None` disables prompt logprobs. `-1` requests the full vocabulary.
+    pub prompt_logprobs: Option<i32>,
    /// Minimum probability threshold for token sampling.
    pub min_p: f32,
    /// Frequency penalty applied by the sampler.
@@ -353,7 +345,6 @@ impl EngineCoreSamplingParams {
            seed: None,
            max_tokens: 65536,
            min_tokens: 0,
-            thinking_token_budget: None,
            logprobs: None,
            prompt_logprobs: None,
            min_p: 0.0,
@@ -150,7 +150,6 @@ fn sample_request_with_id(request_id: &str) -> EngineCoreRequest {
            top_k: 8,
            max_tokens: 32,
            min_tokens: 1,
-            thinking_token_budget: Some(256),
            stop_token_ids: vec![151643],
            eos_token_id: Some(151645),
            all_stop_token_ids: BTreeSet::from([151643, 151645]),
@@ -2503,7 +2502,6 @@ fn python_msgpack_fixtures_match_rust_encoding() {
            seed: None,
            max_tokens: 16,
            min_tokens: 0,
-            thinking_token_budget: None,
            logprobs: None,
            prompt_logprobs: None,
            min_p: 0.0,
@@ -39,7 +39,6 @@ class EngineCoreSamplingParams(msgspec.Struct, dict=True, omit_defaults=True):
    seed: int | None = None
    max_tokens: int = 16
    min_tokens: int = 0
-    thinking_token_budget: int | None = None
    min_p: float = 0.0
    frequency_penalty: float = 0.0
    presence_penalty: float = 0.0
@@ -123,7 +122,6 @@ request = EngineCoreRequest(
        seed=None,
        max_tokens=32,
        min_tokens=1,
-        thinking_token_budget=256,
        min_p=0.0,
        frequency_penalty=0.0,
        presence_penalty=0.0,
@@ -2,13 +2,12 @@ use std::collections::HashMap;
 use std::fmt;
 use std::time::Duration;

-use anyhow::Result;
+use anyhow::{Result, bail};
 use axum::http::{HeaderName, HeaderValue, Method};
 use educe::Educe;
 use serde::Serialize;
 use serde_json::Value;
 use vllm_chat::{ChatTemplateContentFormatOption, ParserSelection, RendererSelection};
-use vllm_engine_core_client::protocol::LogprobsCount;
 use vllm_engine_core_client::{CoordinatorMode as EngineCoreCoordinatorMode, TransportMode};

 /// How the HTTP server obtains its listening socket.
@@ -134,7 +133,7 @@ pub struct Config {
    pub chat_template_content_format: ChatTemplateContentFormatOption,
    /// Optional maximum number of top log probabilities accepted by the
    /// frontend. `None` delegates to the text layer default.
-    pub max_logprobs: Option<LogprobsCount>,
+    pub max_logprobs: Option<i32>,
    /// HTTP/API-server behavior switches.
    pub api_server_options: ApiServerOptions,
    /// CORS settings applied to every HTTP response.
@@ -159,6 +158,15 @@ impl Config {
    pub fn validate(&self) -> Result<()> {
        vllm_chat::validate_parser_overrides(&self.tool_call_parser, &self.reasoning_parser)?;
        self.cors.validate()?;
+        if let Some(max_logprobs) = self.max_logprobs
+            && max_logprobs < -1
+        {
+            bail!(
+                "max_logprobs must be non-negative or -1, got {}",
+                max_logprobs
+            );
+        }
+
        Ok(())
    }

@@ -103,7 +103,6 @@ fn is_request_validation_error(error: &vllm_text::Error) -> bool {
            | vllm_text::Error::EmptyPromptTokenIds { .. }
            | vllm_text::Error::Logprobs(_)
            | vllm_text::Error::OutOfVocab(_)
-            | vllm_text::Error::InvalidThinkingTokenBudget
            // An empty tokenized prompt detected later, at request prepare
            // time, surfaces through the transparent Llm wrapper.
            | vllm_text::Error::Llm(vllm_llm::Error::EmptyPromptTokenIds { .. })
@@ -128,18 +127,6 @@ mod tests {
        assert!(response.error.message.contains("9000"));
    }

-    #[test]
-    fn invalid_thinking_token_budget_maps_to_invalid_request() {
-        let api_error = text_submit_error(
-            "failed to submit completion request",
-            vllm_text::Error::InvalidThinkingTokenBudget,
-        );
-        assert_eq!(api_error.status_code(), StatusCode::BAD_REQUEST);
-        let response = api_error.to_error_response();
-        assert_eq!(response.error.error_type, "invalid_request_error");
-        assert!(response.error.message.contains("thinking_token_budget"));
-    }
-
    #[test]
    fn chat_wrapped_prompt_too_long_maps_to_invalid_request() {
        let error = vllm_chat::Error::Text(vllm_text::Error::PromptTooLong {
@@ -3,7 +3,7 @@

 use tonic::Status;
 use uuid::Uuid;
-use vllm_engine_core_client::protocol::{LogprobsCount, StopReason, StructuredOutputsParams};
+use vllm_engine_core_client::protocol::{StopReason, StructuredOutputsParams};
 use vllm_text::{
    DecodedLogprobs, DecodedPromptLogprobs, FinishReason, Finished, Prompt, SamplingParams,
    TextDecodeOptions, TextRequest,
@@ -202,22 +202,18 @@ fn build_sampling_params(
 /// Map the proto `CandidateTokens` selector to a `(logprobs_count,
 /// logprob_token_ids)` pair.
 ///
-/// - `top_n(k)` → `(Top(k), None)` — return top-k candidates by probability
-/// - `all` → `(All, None)` — return the full vocabulary
+/// - `top_n(k)` → `(k, None)` — return top-k candidates by probability
+/// - `all` → `(-1, None)` — return the full vocabulary
 /// - `token_ids(n)` → `(1, Some(vec of n token ids))` — return logprobs for specific tokens (the
 ///   count `n` is stored in the proto as the number of token IDs that follow, but the actual IDs
 ///   are carried via `logprob_token_ids` on `SamplingParams`)
-/// - absent → `(Top(1), None)` — just the sampled/scored token
-fn candidate_logprob_spec(
-    candidates: Option<&pb::CandidateTokens>,
-) -> (LogprobsCount, Option<Vec<u32>>) {
+/// - absent → `(1, None)` — just the sampled/scored token
+fn candidate_logprob_spec(candidates: Option<&pb::CandidateTokens>) -> (i32, Option<Vec<u32>>) {
    match candidates.and_then(|c| c.select.as_ref()) {
-        Some(pb::candidate_tokens::Select::TopN(n)) => (LogprobsCount::Top(*n), None),
-        Some(pb::candidate_tokens::Select::All(true)) => (LogprobsCount::All, None),
-        Some(pb::candidate_tokens::Select::TokenIds(ids)) => {
-            (LogprobsCount::Top(1), Some(ids.ids.clone()))
-        }
-        _ => (LogprobsCount::Top(1), None),
+        Some(pb::candidate_tokens::Select::TopN(n)) => (*n as i32, None),
+        Some(pb::candidate_tokens::Select::All(true)) => (-1, None),
+        Some(pb::candidate_tokens::Select::TokenIds(ids)) => (1, Some(ids.ids.clone())),
+        _ => (1, None),
    }
 }

@@ -87,7 +87,6 @@ pub(super) fn prepare_generate_request(
 #[cfg(test)]
 mod tests {
    use serde_json::json;
-    use vllm_engine_core_client::protocol::LogprobsCount;
    use vllm_text::Prompt;

    use super::prepare_generate_request;
@@ -133,13 +132,10 @@ mod tests {
            Prompt::TokenIds(vec![11, 22, 33])
        );
        assert_eq!(prepared.text_request.sampling_params.max_tokens, Some(7));
-        assert_eq!(
-            prepared.text_request.sampling_params.logprobs,
-            Some(LogprobsCount::Top(2))
-        );
+        assert_eq!(prepared.text_request.sampling_params.logprobs, Some(2));
        assert_eq!(
            prepared.text_request.sampling_params.prompt_logprobs,
-            Some(LogprobsCount::Top(1))
+            Some(1)
        );
        assert!(prepared.text_request.sampling_params.ignore_eos);
        assert_eq!(prepared.text_request.priority, -3);
@@ -154,33 +150,6 @@ mod tests {
        );
    }

-    #[test]
-    fn prepare_generate_request_forwards_thinking_token_budget() {
-        let request: GenerateRequest = serde_json::from_value(json!({
-            "model": "Qwen/Qwen1.5-0.5B-Chat",
-            "token_ids": [11, 22, 33],
-            "sampling_params": {
-                "thinking_token_budget": 64
-            }
-        }))
-        .expect("parse request");
-
-        let prepared = prepare_generate_request(
-            request,
-            &served(&["Qwen/Qwen1.5-0.5B-Chat"]),
-            ResolvedRequestContext::default(),
-        )
-        .expect("prepare");
-
-        // The raw inference route shares `vllm_text::SamplingParams`, so the
-        // field is carried through to lowering exactly like the OpenAI routes
-        // (normalization/validation then happens in `lower_sampling_params`).
-        assert_eq!(
-            prepared.text_request.sampling_params.thinking_token_budget,
-            Some(64)
-        );
-    }
-
    #[test]
    fn prepare_generate_request_gates_continuous_usage_on_include_usage() {
        let request: GenerateRequest = serde_json::from_value(json!({
@@ -34,6 +34,16 @@ pub(super) fn validate_request_compat(
        );
    }

+    if let Some(prompt_logprobs) = request.sampling_params.prompt_logprobs
+        && prompt_logprobs < 0
+        && prompt_logprobs != -1
+    {
+        bail_invalid_request!(
+            param = "sampling_params",
+            "`prompt_logprobs` must be a non-negative value or -1."
+        );
+    }
+
    Ok(())
 }

@@ -4,7 +4,6 @@ use vllm_chat::{
    ChatMessage as VllmChatMessage, ChatOptions, ChatRequest, ChatTool, ChatToolChoice,
    GenerationPromptMode, SamplingParams,
 };
-use vllm_engine_core_client::protocol::LogprobsCount;

 use super::types::ChatCompletionRequest;
 use super::validate;
@@ -95,7 +94,7 @@ pub(super) fn prepare_chat_request(

    // Auto-enable prompt logprobs for non-streaming echo, matching Python vLLM's
    // behavior.
-    let top_logprobs = request.top_logprobs.unwrap_or(LogprobsCount::Top(0));
+    let top_logprobs = request.top_logprobs.unwrap_or(0);
    let prompt_logprobs = request
        .prompt_logprobs
        .or((request.echo && !request.stream).then_some(top_logprobs));
@@ -116,7 +115,6 @@ pub(super) fn prepare_chat_request(
            seed: request.seed,
            max_tokens: request.max_completion_tokens,
            min_tokens: request.min_tokens,
-            thinking_token_budget: request.thinking_token_budget,
            logprobs: request.logprobs.then_some(top_logprobs),
            prompt_logprobs,
            min_p: request.min_p,
@@ -379,7 +377,6 @@ mod tests {
        ChatTool as VllmChatTool, ChatToolChoice, GenerationPromptMode,
        SamplingParams as VllmSamplingParams,
    };
-    use vllm_engine_core_client::protocol::LogprobsCount;
    use vllm_text::output::TextDecodeOptions;

    use super::prepare_chat_request;
@@ -616,31 +613,6 @@ mod tests {
        assert_eq!(prepared.chat_request.sampling_params, expected);
    }

-    #[test]
-    fn prepare_chat_request_passes_through_thinking_token_budget() {
-        let prepare = |budget: Option<i64>| {
-            prepare_chat_request(
-                ChatCompletionRequest {
-                    thinking_token_budget: budget,
-                    ..base_request()
-                },
-                &served(&["Qwen/Qwen1.5-0.5B-Chat"]),
-                ResolvedRequestContext::default(),
-            )
-            .expect("request is valid")
-            .chat_request
-            .sampling_params
-            .thinking_token_budget
-        };
-
-        // The convert layer forwards the raw value verbatim (including the `-1`
-        // "unlimited" sentinel); normalization/validation happens during
-        // lowering (see `vllm_text::lower`).
-        assert_eq!(prepare(Some(64)), Some(64));
-        assert_eq!(prepare(Some(-1)), Some(-1));
-        assert_eq!(prepare(None), None);
-    }
-
    #[test]
    fn prepare_chat_request_accepts_developer_messages() {
        let request = ChatCompletionRequest {
@@ -969,7 +941,7 @@ mod tests {
        let request = ChatCompletionRequest {
            stream: false,
            logprobs: true,
-            prompt_logprobs: Some(LogprobsCount::Top(2)),
+            prompt_logprobs: Some(2),
            ..base_request()
        };

@@ -982,13 +954,10 @@ mod tests {

        assert!(prepared.options.requested_logprobs);
        assert!(prepared.options.include_prompt_logprobs);
-        assert_eq!(
-            prepared.chat_request.sampling_params.logprobs,
-            Some(LogprobsCount::Top(0))
-        );
+        assert_eq!(prepared.chat_request.sampling_params.logprobs, Some(0));
        assert_eq!(
            prepared.chat_request.sampling_params.prompt_logprobs,
-            Some(LogprobsCount::Top(2))
+            Some(2)
        );
    }

@@ -996,7 +965,7 @@ mod tests {
    fn prepare_chat_request_keeps_prompt_logprobs_independent_from_echo() {
        let request = ChatCompletionRequest {
            logprobs: true,
-            top_logprobs: Some(LogprobsCount::Top(3)),
+            top_logprobs: Some(3),
            echo: true,
            ..base_request()
        };
@@ -1008,10 +977,7 @@ mod tests {
        )
        .expect("request is valid");

-        assert_eq!(
-            prepared.chat_request.sampling_params.logprobs,
-            Some(LogprobsCount::Top(3))
-        );
+        assert_eq!(prepared.chat_request.sampling_params.logprobs, Some(3));
        assert_eq!(prepared.chat_request.sampling_params.prompt_logprobs, None);
        assert!(!prepared.options.include_prompt_logprobs);
    }
@@ -6,7 +6,6 @@ use serde_json::Value;
 use serde_with::SerializeDisplay;
 use validator::Validate;
 use vllm_chat::ReasoningEffort;
-use vllm_engine_core_client::protocol::LogprobsCount;

 use crate::routes::openai::utils::structured_outputs::ResponseFormat;
 use crate::routes::openai::utils::types::{
@@ -45,8 +44,10 @@ pub struct ChatCompletionRequest {
    #[serde(default)]
    pub logprobs: bool,

-    /// Number of most likely tokens to return. `-1` means return full vocab.
-    pub top_logprobs: Option<LogprobsCount>,
+    /// An integer specifying the number of most likely tokens to return
+    /// -1 means return all
+    #[validate(range(min = -1))]
+    pub top_logprobs: Option<i32>,

    /// Deprecated: Replaced by max_completion_tokens
    #[deprecated(note = "Use max_completion_tokens instead")]
@@ -154,8 +155,8 @@ pub struct ChatCompletionRequest {
    /// Truncate prompt tokens to this length
    pub truncate_prompt_tokens: Option<i64>,

-    /// Number of prompt logprobs to return. `-1` means return full vocab.
-    pub prompt_logprobs: Option<LogprobsCount>,
+    /// Number of prompt logprobs to return
+    pub prompt_logprobs: Option<i32>,

    /// Restrict output to these token IDs only
    pub allowed_token_ids: Option<Vec<u32>>,
@@ -164,10 +165,8 @@ pub struct ChatCompletionRequest {
    pub bad_words: Option<Vec<String>>,

    // -------- Extra vLLM Parameters --------
-    /// Token budget for reasoning/thinking. Accepts a non-negative integer, or
-    /// `-1` for unlimited (mirroring the Python frontend, which normalizes `-1`
-    /// to "no budget").
-    pub thinking_token_budget: Option<i64>,
+    /// Token budget for reasoning/thinking
+    pub thinking_token_budget: Option<u32>,

    /// Whether to include reasoning content in the response
    #[serde(default = "default_true")]
@@ -1,7 +1,6 @@
 use super::types::ChatCompletionRequest;
 use crate::error::{ApiError, bail_invalid_request};
 use crate::routes::openai::utils::types::{ChatMessage, Tool, ToolChoice, ToolChoiceValue};
-use vllm_engine_core_client::protocol::LogprobsCount;

 /// Enforce the minimal compatibility contract for the Rust OpenAI server.
 pub(super) fn validate_request_compat(
@@ -31,12 +30,14 @@ pub(super) fn validate_request_compat(
    }

    if let Some(prompt_logprobs) = request.prompt_logprobs {
-        if request.stream
-            && matches!(
-                prompt_logprobs,
-                LogprobsCount::All | LogprobsCount::Top(1..)
-            )
-        {
+        if prompt_logprobs < 0 && prompt_logprobs != -1 {
+            bail_invalid_request!(
+                param = "prompt_logprobs",
+                "prompt_logprobs must be a non-negative value or -1."
+            );
+        }
+
+        if request.stream && (prompt_logprobs > 0 || prompt_logprobs == -1) {
            bail_invalid_request!(
                param = "prompt_logprobs",
                "prompt_logprobs are not available when stream=true."
@@ -107,6 +108,11 @@ pub(super) fn validate_request_compat(
        "truncate_prompt_tokens",
        "truncate_prompt_tokens is not supported.",
    )?;
+    reject_non_default(
+        request.thinking_token_budget.as_ref(),
+        "thinking_token_budget",
+        "thinking_token_budget is not supported.",
+    )?;
    reject_non_default(
        request.media_io_kwargs.as_ref(),
        "media_io_kwargs",
@@ -153,7 +159,6 @@ mod tests {

    use serde_json::json;
    use vllm_chat::ReasoningEffort;
-    use vllm_engine_core_client::protocol::LogprobsCount;

    use super::validate_request_compat;
    use crate::routes::openai::chat_completions::types::ChatCompletionRequest;
@@ -299,7 +304,7 @@ mod tests {
    #[test]
    fn validate_request_compat_rejects_top_logprobs_without_logprobs() {
        let request = ChatCompletionRequest {
-            top_logprobs: Some(LogprobsCount::Top(0)),
+            top_logprobs: Some(0),
            ..base_request()
        };
        assert!(validate_request_compat(&request, &served(&["Qwen/Qwen1.5-0.5B-Chat"])).is_err());
@@ -308,26 +313,26 @@ mod tests {
    #[test]
    fn validate_request_compat_rejects_streaming_prompt_logprobs_requests() {
        let request = ChatCompletionRequest {
-            prompt_logprobs: Some(LogprobsCount::Top(1)),
+            prompt_logprobs: Some(1),
            ..base_request()
        };
        assert!(validate_request_compat(&request, &served(&["Qwen/Qwen1.5-0.5B-Chat"])).is_err());

        let request = ChatCompletionRequest {
-            prompt_logprobs: Some(LogprobsCount::All),
+            prompt_logprobs: Some(-1),
            ..base_request()
        };
        assert!(validate_request_compat(&request, &served(&["Qwen/Qwen1.5-0.5B-Chat"])).is_err());
    }

    #[test]
-    fn chat_request_deserialization_rejects_invalid_prompt_logprobs_value() {
-        let result = serde_json::from_value::<ChatCompletionRequest>(json!({
-            "model": "Qwen/Qwen1.5-0.5B-Chat",
-            "messages": [{"role": "user", "content": "hello"}],
-            "prompt_logprobs": -2
-        }));
-        assert!(result.is_err());
+    fn validate_request_compat_rejects_invalid_prompt_logprobs_value() {
+        let request = ChatCompletionRequest {
+            stream: false,
+            prompt_logprobs: Some(-2),
+            ..base_request()
+        };
+        assert!(validate_request_compat(&request, &served(&["Qwen/Qwen1.5-0.5B-Chat"])).is_err());
    }

    #[test]
@@ -1,4 +1,3 @@
-use vllm_engine_core_client::protocol::LogprobsCount;
 use vllm_text::{SamplingParams, TextDecodeOptions, TextRequest};

 use super::types::CompletionRequest;
@@ -62,7 +61,15 @@ pub(super) fn prepare_completion_request(
        .map(|request| request.lora_name.clone())
        .unwrap_or_else(|| lora_resolution.model_names.first().cloned().unwrap_or_default());

-    let logprobs = request.logprobs.map(LogprobsCount::Top);
+    let logprobs = match request.logprobs {
+        Some(logprobs) => Some(i32::try_from(logprobs).map_err(|_| {
+            ApiError::invalid_request(
+                "`logprobs` must fit within a signed 32-bit integer.".to_string(),
+                Some("logprobs"),
+            )
+        })?),
+        None => None,
+    };
    let prompt_only = request.echo && request.max_tokens == Some(0);
    let prompt_logprobs =
        request.prompt_logprobs.or(if request.echo && (!request.stream || prompt_only) {
@@ -101,7 +108,6 @@ pub(super) fn prepare_completion_request(
            seed: request.seed,
            max_tokens,
            min_tokens: request.min_tokens,
-            thinking_token_budget: request.thinking_token_budget,
            logprobs,
            prompt_logprobs,
            min_p: request.min_p,
@@ -156,7 +162,6 @@ pub(super) fn prepare_completion_request(
 mod tests {
    use axum::http::HeaderMap;
    use serde_json::json;
-    use vllm_engine_core_client::protocol::LogprobsCount;
    use vllm_text::Prompt;

    use super::prepare_completion_request;
@@ -241,10 +246,7 @@ mod tests {
            Prompt::TokenIds(vec![11, 22, 33])
        );
        assert_eq!(prepared.text_request.sampling_params.max_tokens, Some(7));
-        assert_eq!(
-            prepared.text_request.sampling_params.logprobs,
-            Some(LogprobsCount::Top(2))
-        );
+        assert_eq!(prepared.text_request.sampling_params.logprobs, Some(2));
        assert_eq!(prepared.text_request.sampling_params.top_p, Some(0.9));
        assert_eq!(prepared.text_request.sampling_params.top_k, Some(42));
        assert_eq!(prepared.text_request.sampling_params.min_p, Some(0.1));
@@ -264,34 +266,6 @@ mod tests {
        assert!(!prepared.text_request.decode_options.skip_special_tokens);
    }

-    #[test]
-    fn prepare_completion_request_passes_through_thinking_token_budget() {
-        let prepare = |budget: serde_json::Value| {
-            let request: CompletionRequest = serde_json::from_value(json!({
-                "model": "Qwen/Qwen1.5-0.5B-Chat",
-                "prompt": "hello",
-                "thinking_token_budget": budget,
-            }))
-            .expect("parse request");
-            prepare_completion_request(
-                request,
-                &served(&["Qwen/Qwen1.5-0.5B-Chat"]),
-                ResolvedRequestContext::default(),
-            )
-            .expect("prepare")
-            .text_request
-            .sampling_params
-            .thinking_token_budget
-        };
-
-        // The convert layer forwards the raw value verbatim (including the `-1`
-        // "unlimited" sentinel); normalization/validation happens during
-        // lowering (see `vllm_text::lower`).
-        assert_eq!(prepare(json!(64)), Some(64));
-        assert_eq!(prepare(json!(-1)), Some(-1));
-        assert_eq!(prepare(json!(null)), None);
-    }
-
    #[test]
    fn prepare_completion_request_maps_stream_usage_and_token_format_options() {
        let request: CompletionRequest = serde_json::from_value(json!({
@@ -407,13 +381,10 @@ mod tests {
        .expect("prepare");

        assert!(prepared.options.prompt_only);
-        assert_eq!(
-            prepared.text_request.sampling_params.logprobs,
-            Some(LogprobsCount::Top(3))
-        );
+        assert_eq!(prepared.text_request.sampling_params.logprobs, Some(3));
        assert_eq!(
            prepared.text_request.sampling_params.prompt_logprobs,
-            Some(LogprobsCount::Top(3))
+            Some(3)
        );
    }

@@ -435,13 +406,10 @@ mod tests {
        )
        .expect("prepare");

-        assert_eq!(
-            prepared.text_request.sampling_params.logprobs,
-            Some(LogprobsCount::Top(3))
-        );
+        assert_eq!(prepared.text_request.sampling_params.logprobs, Some(3));
        assert_eq!(
            prepared.text_request.sampling_params.prompt_logprobs,
-            Some(LogprobsCount::Top(3))
+            Some(3)
        );
    }

@@ -482,13 +450,10 @@ mod tests {
            ResolvedRequestContext::default(),
        )
        .expect("prepare");
-        assert_eq!(
-            prepared.text_request.sampling_params.logprobs,
-            Some(LogprobsCount::Top(1))
-        );
+        assert_eq!(prepared.text_request.sampling_params.logprobs, Some(1));
        assert_eq!(
            prepared.text_request.sampling_params.prompt_logprobs,
-            Some(LogprobsCount::Top(2))
+            Some(2)
        );
    }

@@ -3,7 +3,6 @@ use std::collections::HashMap;
 use serde::{Deserialize, Serialize};
 use serde_json::{Map, Value};
 use validator::Validate;
-use vllm_engine_core_client::protocol::LogprobsCount;
 use vllm_text::Prompt;

 use crate::routes::openai::utils::types::{
@@ -132,8 +131,8 @@ pub struct CompletionRequest {
    /// Restrict output to these token IDs only
    pub allowed_token_ids: Option<Vec<u32>>,

-    /// Number of prompt logprobs to return. `-1` means return full vocab.
-    pub prompt_logprobs: Option<LogprobsCount>,
+    /// Number of prompt logprobs to return
+    pub prompt_logprobs: Option<i32>,

    // -------- Extra vLLM Parameters --------
    /// Whether to add special tokens (e.g. BOS) to the prompt
@@ -147,11 +146,6 @@ pub struct CompletionRequest {
    /// Additional kwargs for structured outputs
    pub structured_outputs: Option<Value>,

-    /// Token budget for reasoning/thinking. Accepts a non-negative integer, or
-    /// `-1` for unlimited (mirroring the Python frontend, which normalizes `-1`
-    /// to "no budget").
-    pub thinking_token_budget: Option<i64>,
-
    /// Request scheduling priority (lower means earlier; default 0)
    pub priority: Option<i32>,

@@ -1,4 +1,3 @@
-use vllm_engine_core_client::protocol::LogprobsCount;
 use vllm_text::Prompt;

 use super::types::CompletionRequest;
@@ -45,18 +44,29 @@ pub(super) fn validate_request_compat(
        bail_invalid_request!(param = "suffix", "suffix is not supported.");
    }

+    if let Some(logprobs) = request.logprobs
+        && logprobs > i32::MAX as u32
+    {
+        bail_invalid_request!(
+            param = "logprobs",
+            "`logprobs` must fit within a signed 32-bit integer."
+        );
+    }
+
    if let Some(prompt_logprobs) = request.prompt_logprobs {
-        if request.stream
-            && matches!(
-                prompt_logprobs,
-                LogprobsCount::All | LogprobsCount::Top(1..)
-            )
-        {
+        if request.stream && (prompt_logprobs > 0 || prompt_logprobs == -1) {
            bail_invalid_request!(
                param = "prompt_logprobs",
                "`prompt_logprobs` are not available when `stream=true`."
            );
        }
+
+        if prompt_logprobs < 0 && prompt_logprobs != -1 {
+            bail_invalid_request!(
+                param = "prompt_logprobs",
+                "`prompt_logprobs` must be a non-negative value or -1."
+            );
+        }
    }

    if request.use_beam_search {
@@ -91,7 +101,6 @@ pub(super) fn validate_request_compat(
 #[cfg(test)]
 mod tests {
    use serde_json::json;
-    use vllm_engine_core_client::protocol::LogprobsCount;

    use super::validate_request_compat;
    use crate::routes::openai::completions::types::CompletionRequest;
@@ -141,7 +150,7 @@ mod tests {
    #[test]
    fn validate_request_compat_rejects_streaming_prompt_logprobs() {
        let request = CompletionRequest {
-            prompt_logprobs: Some(LogprobsCount::Top(1)),
+            prompt_logprobs: Some(1),
            ..base_request()
        };
        assert!(
@@ -153,7 +162,7 @@ mod tests {
    fn validate_request_compat_accepts_non_stream_prompt_logprobs() {
        let request = CompletionRequest {
            stream: false,
-            prompt_logprobs: Some(LogprobsCount::All),
+            prompt_logprobs: Some(-1),
            ..base_request()
        };
        assert!(
@@ -2,7 +2,6 @@ pub mod hf;

 use std::sync::Arc;

-use vllm_engine_core_client::protocol::LogprobsCount;
 use vllm_tokenizer::DynTokenizer;

 use crate::error::Result;
@@ -27,7 +26,9 @@ pub struct SamplingLimits {
    /// Runtime context window size reported by the engine startup handshake.
    pub max_model_len: u32,
    /// Maximum number of top log probabilities accepted by this frontend.
-    pub max_logprobs: LogprobsCount,
+    ///
+    /// `-1` means allowing requests up to the model vocabulary size.
+    pub max_logprobs: i32,

    /// Model vocabulary size from the model config, used to bound generated
    /// token IDs and logits-domain sampling controls.
@@ -40,7 +41,7 @@ pub struct SamplingLimits {
 impl SamplingLimits {
    /// Original Python definition:
    /// <https://github.com/vllm-project/vllm/blob/b5adb027ad03c29b46181752ba3b1cb84eff1dd4/vllm/config/model.py#L216-L220>
-    pub const DEFAULT_MAX_LOGPROBS: LogprobsCount = LogprobsCount::Top(20);
+    pub const DEFAULT_MAX_LOGPROBS: i32 = 20;
    /// Original Python definition:
    /// <https://github.com/vllm-project/vllm/blob/b5adb027ad03c29b46181752ba3b1cb84eff1dd4/vllm/sampling_params.py#L30-L32>
    pub const MAX_LOGPROB_TOKEN_IDS: usize = 128;
@@ -20,8 +20,6 @@ pub enum Error {
    Logprobs(#[from] LogprobsError),
    #[error(transparent)]
    OutOfVocab(#[from] OutOfVocabError),
-    #[error("`thinking_token_budget` must be a non-negative integer or -1 for unlimited.")]
-    InvalidThinkingTokenBudget,
    #[error("text request stream `{request_id}` closed before terminal output")]
    StreamClosedBeforeTerminalOutput { request_id: String },
    #[error(transparent)]
@@ -19,7 +19,6 @@ pub use output::{
 pub use request::{Prompt, SamplingParams, TextRequest};
 use trait_set::trait_set;
 use vllm_engine_core_client::EngineCoreClient;
-use vllm_engine_core_client::protocol::LogprobsCount;
 pub use vllm_llm::FinishReason;
 use vllm_llm::{GenerateOutputStream, Llm};
 use vllm_tokenizer::DynTokenizer;
@@ -49,7 +48,7 @@ pub struct TextLlm {
    /// Runtime context window size reported by the engine startup handshake.
    max_model_len: u32,
    /// Maximum number of top log probabilities accepted by this text facade.
-    max_logprobs: LogprobsCount,
+    max_logprobs: i32,
 }

 impl TextLlm {
@@ -69,7 +68,7 @@ impl TextLlm {
    }

    /// Override the maximum accepted logprobs count.
-    pub fn with_max_logprobs(mut self, max_logprobs: Option<LogprobsCount>) -> Self {
+    pub fn with_max_logprobs(mut self, max_logprobs: Option<i32>) -> Self {
        if let Some(max_logprobs) = max_logprobs {
            self.max_logprobs = max_logprobs;
        }
@@ -87,7 +87,6 @@ pub fn lower_sampling_params(
        seed,
        max_tokens,
        min_tokens,
-        thinking_token_budget,
        logprobs,
        prompt_logprobs,
        min_p,
@@ -129,7 +128,6 @@ pub fn lower_sampling_params(
        prompt_len,
    )?;
    let min_tokens = min_tokens.unwrap_or(0);
-    let thinking_token_budget = normalize_thinking_token_budget(thinking_token_budget)?;
    let frequency_penalty = frequency_penalty.unwrap_or(0.0);
    let presence_penalty = presence_penalty.unwrap_or(0.0);

@@ -151,7 +149,6 @@ pub fn lower_sampling_params(
        seed,
        max_tokens,
        min_tokens,
-        thinking_token_budget,
        logprobs,
        prompt_logprobs,
        min_p,
@@ -173,21 +170,6 @@ pub fn lower_sampling_params(
    Ok(params)
 }

-/// Normalize the user-facing `thinking_token_budget` into the engine value.
-///
-/// Mirrors Python's `validate_thinking_token_budget`
-/// (<https://github.com/vllm-project/vllm/blob/ecf9d83520eb217401b47d8a5451a27c5231b8c2/vllm/sampling_params.py#L35-L55>):
-/// `None` and the `-1` "unlimited" sentinel both map to `None`; any other
-/// negative value is rejected; non-negative values pass through unchanged. Like
-/// Python's `int`, no upper bound is imposed.
-fn normalize_thinking_token_budget(value: Option<i64>) -> Result<Option<u64>> {
-    match value {
-        None | Some(-1) => Ok(None),
-        Some(budget) if budget >= 0 => Ok(Some(budget as u64)),
-        Some(_) => Err(Error::InvalidThinkingTokenBudget),
-    }
-}
-
 /// Convert bad-word strings into token-ID sequences, following the Python vLLM
 /// logic in `SamplingParams.update_from_tokenizer()`.
 ///
@@ -269,7 +251,6 @@ mod tests {
    use std::collections::{BTreeSet, HashMap};

    use serial_test::file_serial;
-    use vllm_engine_core_client::protocol::LogprobsCount;

    use super::*;
    use crate::backend::hf::HfTextBackend;
@@ -385,36 +366,6 @@ mod tests {
        )
    }

-    #[test]
-    fn lower_sampling_params_normalizes_thinking_token_budget() {
-        let lower = |budget: Option<i64>| {
-            lower_sampling_params_with_limits(
-                SamplingParams {
-                    thinking_token_budget: budget,
-                    ..SamplingParams::default()
-                },
-                sample_sampling_limits(),
-            )
-        };
-
-        // Non-negative budgets (including 0) pass through unchanged.
-        assert_eq!(lower(Some(256)).unwrap().thinking_token_budget, Some(256));
-        assert_eq!(lower(Some(0)).unwrap().thinking_token_budget, Some(0));
-        // `None` and the `-1` "unlimited" sentinel both disable the budget.
-        assert_eq!(lower(None).unwrap().thinking_token_budget, None);
-        assert_eq!(lower(Some(-1)).unwrap().thinking_token_budget, None);
-        // No upper bound is imposed, matching Python's `int`.
-        assert_eq!(
-            lower(Some(i64::from(u32::MAX) + 1)).unwrap().thinking_token_budget,
-            Some(u64::from(u32::MAX) + 1)
-        );
-        // Other negatives are rejected.
-        assert!(matches!(
-            lower(Some(-2)),
-            Err(Error::InvalidThinkingTokenBudget)
-        ));
-    }
-
    #[test]
    fn lower_text_request_applies_python_style_eos_hints() {
        let prepared = lower_text_request(
@@ -435,7 +386,6 @@ mod tests {
                seed: None,
                max_tokens: 999997,
                min_tokens: 0,
-                thinking_token_budget: None,
                logprobs: None,
                prompt_logprobs: None,
                min_p: 0.0,
@@ -487,7 +437,6 @@ mod tests {
                seed: None,
                max_tokens: 999997,
                min_tokens: 0,
-                thinking_token_budget: None,
                logprobs: None,
                prompt_logprobs: None,
                min_p: 0.0,
@@ -618,7 +567,6 @@ mod tests {
                seed: None,
                max_tokens: 40957,
                min_tokens: 0,
-                thinking_token_budget: None,
                logprobs: None,
                prompt_logprobs: None,
                min_p: 0.0,
@@ -680,7 +628,6 @@ mod tests {
                seed: None,
                max_tokens: 999997,
                min_tokens: 0,
-                thinking_token_budget: None,
                logprobs: None,
                prompt_logprobs: None,
                min_p: 0.0,
@@ -750,7 +697,6 @@ mod tests {
                seed: None,
                max_tokens: 32,
                min_tokens: 2,
-                thinking_token_budget: None,
                logprobs: None,
                prompt_logprobs: None,
                min_p: 0.1,
@@ -775,8 +721,8 @@ mod tests {
    #[test]
    fn lower_sampling_params_passes_logprobs_fields_through() {
        let sampling_params = SamplingParams {
-            logprobs: Some(LogprobsCount::Top(3)),
-            prompt_logprobs: Some(LogprobsCount::All),
+            logprobs: Some(3),
+            prompt_logprobs: Some(-1),
            ..Default::default()
        };

@@ -793,7 +739,7 @@ mod tests {
                default_max_tokens: None,
            },
            SamplingLimits {
-                max_logprobs: LogprobsCount::All,
+                max_logprobs: -1,
                ..sample_sampling_limits()
            },
            3,
@@ -801,15 +747,15 @@ mod tests {
        )
        .unwrap();

-        assert_eq!(params.logprobs, Some(LogprobsCount::Top(3)));
-        assert_eq!(params.prompt_logprobs, Some(LogprobsCount::All));
+        assert_eq!(params.logprobs, Some(3));
+        assert_eq!(params.prompt_logprobs, Some(-1));
    }

    #[test]
    fn lower_sampling_params_rejects_full_vocab_logprobs_over_default_cap() {
        let error = lower_sampling_params_with_limits(
            SamplingParams {
-                logprobs: Some(LogprobsCount::All),
+                logprobs: Some(-1),
                ..Default::default()
            },
            sample_sampling_limits(),
@@ -830,24 +776,24 @@ mod tests {
    fn lower_sampling_params_expands_full_vocab_logprobs_from_model_vocab() {
        let params = lower_sampling_params_with_limits(
            SamplingParams {
-                logprobs: Some(LogprobsCount::All),
+                logprobs: Some(-1),
                ..Default::default()
            },
            SamplingLimits {
-                max_logprobs: LogprobsCount::Top(1500),
+                max_logprobs: 1500,
                ..sample_sampling_limits()
            },
        )
        .unwrap();

-        assert_eq!(params.logprobs, Some(LogprobsCount::All));
+        assert_eq!(params.logprobs, Some(-1));
    }

    #[test]
    fn lower_sampling_params_rejects_invalid_logprob_token_ids() {
        let error = lower_sampling_params_with_limits(
            SamplingParams {
-                logprobs: Some(LogprobsCount::Top(1)),
+                logprobs: Some(1),
                logprob_token_ids: Some(vec![1000]),
                ..Default::default()
            },
@@ -983,7 +929,6 @@ mod tests {
                seed: None,
                max_tokens: 128,
                min_tokens: 0,
-                thinking_token_budget: None,
                logprobs: None,
                prompt_logprobs: None,
                min_p: 0.1,
@@ -1,14 +1,15 @@
 //! Python-compatible validation for logprobs sampling params.
 //!
-//! `All` is expanded only for bounds checks. The original request values are
+//! `-1` is expanded only for bounds checks. The original request values are
 //! passed through to engine-core.

 use crate::backend::SamplingLimits;
 use thiserror::Error;
-use vllm_engine_core_client::protocol::LogprobsCount;

 #[derive(Debug, Error)]
 pub enum LogprobsError {
+    #[error("{parameter} must be non-negative or -1, got {value}")]
+    InvalidCount { parameter: &'static str, value: i32 },
    #[error(
        "requested {parameter} of {requested}, which is greater than max allowed: {max_allowed}"
    )]
@@ -29,21 +30,19 @@ pub enum LogprobsError {
        "when both logprobs and logprob_token_ids are set, logprobs must equal \
         len(logprob_token_ids). Got logprobs={logprobs}, len(logprob_token_ids)={num_token_ids}."
    )]
-    TokenIdsMismatch {
-        logprobs: LogprobsCount,
-        num_token_ids: usize,
-    },
+    TokenIdsMismatch { logprobs: i32, num_token_ids: usize },
 }

 /// Validate logprobs count sampling parameters.
 pub(super) fn validate_logprobs(
-    logprobs: Option<LogprobsCount>,
-    prompt_logprobs: Option<LogprobsCount>,
+    logprobs: Option<i32>,
+    prompt_logprobs: Option<i32>,
    logprob_token_ids: Option<&[u32]>,
    sampling_limits: SamplingLimits,
 ) -> Result<(), LogprobsError> {
    let vocab_size = sampling_limits.model_vocab_size;
-    let max_logprobs = sampling_limits.max_logprobs.expanded(vocab_size);
+    let max_logprobs =
+        normalize_logprobs_count(sampling_limits.max_logprobs, vocab_size, "max_logprobs")?;

    validate_logprobs_count(logprobs, max_logprobs, vocab_size, "logprobs")?;
    validate_logprobs_count(prompt_logprobs, max_logprobs, vocab_size, "prompt_logprobs")?;
@@ -51,7 +50,7 @@ pub(super) fn validate_logprobs(
 }

 fn validate_logprobs_count(
-    requested: Option<LogprobsCount>,
+    requested: Option<i32>,
    max_logprobs: usize,
    vocab_size: usize,
    parameter: &'static str,
@@ -60,7 +59,7 @@ fn validate_logprobs_count(
        return Ok(());
    };

-    let requested = requested.expanded(vocab_size);
+    let requested = normalize_logprobs_count(requested, vocab_size, parameter)?;
    if requested > max_logprobs {
        return Err(LogprobsError::TooManyCount {
            parameter,
@@ -73,7 +72,7 @@ fn validate_logprobs_count(
 }

 pub(super) fn validate_logprob_token_ids(
-    logprobs: Option<LogprobsCount>,
+    logprobs: Option<i32>,
    logprob_token_ids: Option<&[u32]>,
 ) -> Result<(), LogprobsError> {
    let Some(logprob_token_ids) = logprob_token_ids else {
@@ -89,7 +88,7 @@ pub(super) fn validate_logprob_token_ids(
    }

    if let Some(logprobs) = logprobs
-        && logprobs != LogprobsCount::Top(n as u32)
+        && logprobs != n as i32
    {
        return Err(LogprobsError::TokenIdsMismatch {
            logprobs,
@@ -99,3 +98,15 @@ pub(super) fn validate_logprob_token_ids(

    Ok(())
 }
+
+fn normalize_logprobs_count(
+    value: i32,
+    vocab_size: usize,
+    parameter: &'static str,
+) -> Result<usize, LogprobsError> {
+    match value {
+        -1 => Ok(vocab_size),
+        value if value < 0 => Err(LogprobsError::InvalidCount { parameter, value }),
+        value => Ok(value as usize),
+    }
+}
@@ -309,7 +309,7 @@ fn matches_stop_string(stops: &[String], output: &str, new_bytes: usize) -> Opti
        .find_map(|(ss_idx, (ss, len, start_off))| {
            output[start_off..]
                .windows(len)
-                .position(|w| w == ss)
+                .rposition(|w| w == ss)
                .map(|pos| (ss_idx, start_off + pos))
        })
 }
@@ -562,13 +562,6 @@ mod tests {
        assert_eq!(result, Some((0, 4)));
    }

-    #[test]
-    fn stop_string_matches_leftmost_with_multiple_new_bytes() {
-        let stops = vec!["\n".to_string()];
-        let result = matches_stop_string(&stops, "Answer\n\n", 2);
-        assert_eq!(result, Some((0, 6)));
-    }
-
    #[test]
    fn stop_string_matches_at_beginning() {
        let stops = vec!["say".to_string()];
@@ -3,9 +3,9 @@ use std::collections::HashMap;
 use enum_as_inner::EnumAsInner;
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
+use vllm_engine_core_client::protocol::StructuredOutputsParams;
 use vllm_engine_core_client::protocol::lora::LoraRequest;
 use vllm_engine_core_client::protocol::multimodal::MmFeatures;
-use vllm_engine_core_client::protocol::{LogprobsCount, StructuredOutputsParams};

 use crate::error::{Error, Result};
 use crate::output::TextDecodeOptions;
@@ -56,20 +56,14 @@ pub struct SamplingParams {
    pub max_tokens: Option<u32>,
    /// Minimum number of tokens to generate before EOS or stop-token handling.
    pub min_tokens: Option<u32>,
-    /// Maximum number of reasoning ("thinking") tokens to emit before the
-    /// reasoning section is force-closed. `None` or the user-facing `-1`
-    /// "unlimited" sentinel both disable the budget. The raw value is carried
-    /// here; `-1` is normalized to `None` (and other negatives rejected) during
-    /// lowering (see `lower_sampling_params`).
-    pub thinking_token_budget: Option<i64>,
    /// Number of log probabilities to return per generated token.
    ///
-    /// `None` disables sample logprobs.
-    pub logprobs: Option<LogprobsCount>,
+    /// `None` disables sample logprobs. `-1` requests the full vocabulary.
+    pub logprobs: Option<i32>,
    /// Number of log probabilities to return per prompt token.
    ///
-    /// `None` disables prompt logprobs.
-    pub prompt_logprobs: Option<LogprobsCount>,
+    /// `None` disables prompt logprobs. `-1` requests the full vocabulary.
+    pub prompt_logprobs: Option<i32>,
    /// Minimum probability threshold for token sampling. `None` means no
    /// explicit user override.
    pub min_p: Option<f32>,
@@ -122,7 +116,6 @@ impl Default for SamplingParams {
            seed: None,
            max_tokens: None,
            min_tokens: None,
-            thinking_token_budget: None,
            logprobs: None,
            prompt_logprobs: None,
            min_p: None,
--- a/Show More
+++ b/Show More
Author	SHA1	Message	Date
Andreas KaratzasandGitHub	78d2334aab	Merge branch 'main' into codex/rocm-artifact-tensorizer	2026-06-18 14:28:51 -05:00
Andreas Karatzas	7b375c8502	Fix ROCm artifact test dependencies Signed-off-by: Andreas Karatzas <akaratza@amd.com>	2026-06-04 21:30:21 -05:00