vllm - ✅(Solved) Fix [Bug][ROCm]: Step3.5 Flash MTP init error [2 pull requests, 4 comments, 3 participants]

Official PRs (…)
ON THIS PAGE

Recommended Tools

×6

Utilities matched from this issue’s tags and category — try them while you read without losing context.

GitHub issue graph ai analysis

Paste a GitHub issue URL. We fetch that issue, discover linked issues from bodies/comments/timeline, collect linked pull requests, and produce a structured English report.

The report is written in English Markdown for sharing and archival.

Helpful · Quick feedback

Loading…
GitHub stats
vllm-project/vllm#38498Fetched 2026-04-08 01:48:59
View on GitHub
Comments
4
Participants
3
Timeline
20
Reactions
0
Timeline (top)
commented ×4mentioned ×4subscribed ×4labeled ×2

Error Message

(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] WorkerProc hit an exception. (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] Traceback (most recent call last): (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/multiproc_executor.py", line 875, in worker_busy_loop (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] output = func(*args, **kwargs) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] ^^^^^^^^^^^^^^^^^^^^^ (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/worker_base.py", line 320, in initialize_from_config (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.worker.initialize_from_config(kv_cache_config) # type: ignore (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] return func(*args, **kwargs) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] ^^^^^^^^^^^^^^^^^^^^^ (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 481, in initialize_from_config (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.model_runner.initialize_kv_cache(kv_cache_config) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 6071, in initialize_kv_cache (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.initialize_metadata_builders(kv_cache_config, kernel_block_sizes) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5557, in initialize_metadata_builders (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.drafter.initialize_attn_backend(kv_cache_config, kernel_block_sizes) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1596, in initialize_attn_backend (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.validate_same_kv_cache_group(kv_cache_config) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1570, in validate_same_kv_cache_group (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] len( (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] AssertionError: All drafting layers should belong to the same kv cache group (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] Traceback (most recent call last): (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/multiproc_executor.py", line 875, in worker_busy_loop (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] output = func(*args, **kwargs) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] ^^^^^^^^^^^^^^^^^^^^^ (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/worker_base.py", line 320, in initialize_from_config (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.worker.initialize_from_config(kv_cache_config) # type: ignore (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] return func(*args, **kwargs) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] ^^^^^^^^^^^^^^^^^^^^^ (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 481, in initialize_from_config (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.model_runner.initialize_kv_cache(kv_cache_config) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 6071, in initialize_kv_cache (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.initialize_metadata_builders(kv_cache_config, kernel_block_sizes) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5557, in initialize_metadata_builders (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.drafter.initialize_attn_backend(kv_cache_config, kernel_block_sizes) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1596, in initialize_attn_backend (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] self.validate_same_kv_cache_group(kv_cache_config) (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1570, in validate_same_kv_cache_group (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] len( (Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] AssertionError: All drafting layers should belong to the same kv cache group

Fix Action

Fix / Workaround

============================== CPU Info

Architecture: x86_64 CPU op-mode(s): 32-bit, 64-bit Address sizes: 52 bits physical, 57 bits virtual Byte Order: Little Endian CPU(s): 384 On-line CPU(s) list: 0-383 Vendor ID: AuthenticAMD BIOS Vendor ID: Advanced Micro Devices, Inc. Model name: AMD EPYC 9K84 96-Core Processor BIOS Model name: AMD EPYC 9K84 96-Core Processor
CPU family: 25 Model: 17 Thread(s) per core: 2 Core(s) per socket: 96 Socket(s): 2 Stepping: 1 Frequency boost: enabled CPU max MHz: 2600.0000 CPU min MHz: 1500.0000 BogoMIPS: 5200.50 Flags: fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush mmx fxsr sse sse2 ht syscall nx mmxext fxsr_opt pdpe1gb rdtscp lm constant_tsc rep_good nopl nonstop_tsc cpuid extd_apicid aperfmperf pni pclmulqdq monitor ssse3 fma cx16 pcid sse4_1 sse4_2 x2apic movbe popcnt aes xsave avx f16c rdrand lahf_lm cmp_legacy svm extapic cr8_legacy abm sse4a misalignsse 3dnowprefetch osvw ibs skinit wdt tce topoext perfctr_core perfctr_nb bpext perfctr_llc mwaitx cpb cat_l3 cdp_l3 invpcid_single hw_pstate ssbd mba ibrs ibpb stibp vmmcall fsgsbase bmi1 avx2 smep bmi2 erms invpcid cqm rdt_a avx512f avx512dq rdseed adx smap avx512ifma clflushopt clwb avx512cd sha_ni avx512bw avx512vl xsaveopt xsavec xgetbv1 xsaves cqm_llc cqm_occup_llc cqm_mbm_total cqm_mbm_local avx512_bf16 clzero irperf xsaveerptr wbnoinvd arat npt lbrv svm_lock nrip_save tsc_scale vmcb_clean flushbyasid decodeassists pausefilter pfthreshold avic v_vmsave_vmload vgif avx512vbmi umip pku ospke avx512_vbmi2 gfni vaes vpclmulqdq avx512_vnni avx512_bitalg avx512_vpopcntdq la57 rdpid overflow_recov succor smca fsrm flush_l1d Virtualization: AMD-V L1d cache: 6 MiB (192 instances) L1i cache: 6 MiB (192 instances) L2 cache: 192 MiB (192 instances) L3 cache: 768 MiB (24 instances) NUMA node(s): 2 NUMA node0 CPU(s): 0-95,192-287 NUMA node1 CPU(s): 96-191,288-383 Vulnerability Itlb multihit: Not affected Vulnerability L1tf: Not affected Vulnerability Mds: Not affected Vulnerability Meltdown: Not affected Vulnerability Spec store bypass: Mitigation; Speculative Store Bypass disabled via prctl and seccomp Vulnerability Spectre v1: Mitigation; usercopy/swapgs barriers and __user pointer sanitization Vulnerability Spectre v2: Mitigation; Full AMD retpoline, IBPB conditional, IBRS_FW, STIBP always-on, RSB filling Vulnerability Srbds: Not affected Vulnerability Tsx async abort: Not affected

PR fix notes

PR #39110: [Core] Disable HMA for eagle/MTP with sliding window models

Description (problem / solution / changelog)

Step-3.5-Flash uses interleaved sliding window and full attention layers. When the hybrid KV cache manager (HMA) is enabled, it splits these into separate KV cache groups. The eagle/MTP drafter then fails during validate_same_kv_cache_group():

AssertionError: All drafting layers should belong to the same kv cache group

There's already a guard for chunked local attention + eagle at line 1233, but no equivalent for sliding window + eagle. This adds one — when speculative decoding is used with a model that has sliding_window set, HMA is auto-disabled so all layers stay in one KV cache group.

The reporter confirmed that manually setting need_disable_hybrid_kv_cache_manager = True resolves the crash.

Fixes #38498

Signed-off-by: Bortlesboat [email protected]

Changed files

  • vllm/config/vllm.py (modified, +12/-0)

PR #39376: [Core] Disable HMA for eagle/MTP with sliding window models

Description (problem / solution / changelog)

Models like Step-3.5-Flash interleave sliding window and full attention layers. When HMA is enabled it splits them into separate KV cache groups, causing validate_same_kv_cache_group() in the eagle/MTP drafter to fail:

AssertionError: All drafting layers should belong to the same kv cache group

This adds a guard to auto-disable HMA when eagle/MTP speculative decoding is combined with a sliding window model, following the same pattern as the existing guard for chunked local attention + eagle at line 1245.

Fixes #38498

Duplicate check: No existing open PR addresses this. Previous PRs and issues discuss the symptom but no fix was merged.

Tests: Verified the guard logic matches the existing chunked_prefill_enabled pattern; unit tests for VllmConfig pass with VLLM_USE_PRECOMPILED=1 uv pip install -e ..

AI assistance: Authored with Claude Code assistance.

cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @jataylo @hongxiayang @naromero77amd

Changed files

  • vllm/config/vllm.py (modified, +12/-0)

Code Example

Collecting environment information...
==============================
        System Info
==============================
OS                           : Ubuntu 22.04.5 LTS (x86_64)
GCC version                  : (Ubuntu 11.4.0-1ubuntu1~22.04.2) 11.4.0
Clang version                : 20.0.0git (https://github.com/RadeonOpenCompute/llvm-project roc-7.0.0 25314 f4087f6b428f0e6f575ebac8a8a724dab123d06e)
CMake version                : version 3.31.10
Libc version                 : glibc-2.35

==============================
       PyTorch Info
==============================
PyTorch version              : 2.9.1+git8907517
Is debug build               : False
CUDA used to build PyTorch   : N/A
ROCM used to build PyTorch   : 7.0.51831-a3e329ad8

==============================
      Python Environment
==============================
Python version               : 3.12.12 (main, Oct 10 2025, 08:52:57) [GCC 11.4.0] (64-bit runtime)
Python platform              : Linux-5.4.119-19.0009.60-x86_64-with-glibc2.35

==============================
       CUDA / GPU Info
==============================
Is CUDA available            : True
CUDA runtime version         : Could not collect
CUDA_MODULE_LOADING set to   : 
GPU models and configuration :  (gfx942:sramecc+:xnack-)
Nvidia driver version        : Could not collect
cuDNN version                : Could not collect
HIP runtime version          : 7.0.51831
MIOpen runtime version       : 3.5.0
Is XNNPACK available         : True

==============================
          CPU Info
==============================
Architecture:                    x86_64
CPU op-mode(s):                  32-bit, 64-bit
Address sizes:                   52 bits physical, 57 bits virtual
Byte Order:                      Little Endian
CPU(s):                          384
On-line CPU(s) list:             0-383
Vendor ID:                       AuthenticAMD
BIOS Vendor ID:                  Advanced Micro Devices, Inc.
Model name:                      AMD EPYC 9K84 96-Core Processor
BIOS Model name:                 AMD EPYC 9K84 96-Core Processor                
CPU family:                      25
Model:                           17
Thread(s) per core:              2
Core(s) per socket:              96
Socket(s):                       2
Stepping:                        1
Frequency boost:                 enabled
CPU max MHz:                     2600.0000
CPU min MHz:                     1500.0000
BogoMIPS:                        5200.50
Flags:                           fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush mmx fxsr sse sse2 ht syscall nx mmxext fxsr_opt pdpe1gb rdtscp lm constant_tsc rep_good nopl nonstop_tsc cpuid extd_apicid aperfmperf pni pclmulqdq monitor ssse3 fma cx16 pcid sse4_1 sse4_2 x2apic movbe popcnt aes xsave avx f16c rdrand lahf_lm cmp_legacy svm extapic cr8_legacy abm sse4a misalignsse 3dnowprefetch osvw ibs skinit wdt tce topoext perfctr_core perfctr_nb bpext perfctr_llc mwaitx cpb cat_l3 cdp_l3 invpcid_single hw_pstate ssbd mba ibrs ibpb stibp vmmcall fsgsbase bmi1 avx2 smep bmi2 erms invpcid cqm rdt_a avx512f avx512dq rdseed adx smap avx512ifma clflushopt clwb avx512cd sha_ni avx512bw avx512vl xsaveopt xsavec xgetbv1 xsaves cqm_llc cqm_occup_llc cqm_mbm_total cqm_mbm_local avx512_bf16 clzero irperf xsaveerptr wbnoinvd arat npt lbrv svm_lock nrip_save tsc_scale vmcb_clean flushbyasid decodeassists pausefilter pfthreshold avic v_vmsave_vmload vgif avx512vbmi umip pku ospke avx512_vbmi2 gfni vaes vpclmulqdq avx512_vnni avx512_bitalg avx512_vpopcntdq la57 rdpid overflow_recov succor smca fsrm flush_l1d
Virtualization:                  AMD-V
L1d cache:                       6 MiB (192 instances)
L1i cache:                       6 MiB (192 instances)
L2 cache:                        192 MiB (192 instances)
L3 cache:                        768 MiB (24 instances)
NUMA node(s):                    2
NUMA node0 CPU(s):               0-95,192-287
NUMA node1 CPU(s):               96-191,288-383
Vulnerability Itlb multihit:     Not affected
Vulnerability L1tf:              Not affected
Vulnerability Mds:               Not affected
Vulnerability Meltdown:          Not affected
Vulnerability Spec store bypass: Mitigation; Speculative Store Bypass disabled via prctl and seccomp
Vulnerability Spectre v1:        Mitigation; usercopy/swapgs barriers and __user pointer sanitization
Vulnerability Spectre v2:        Mitigation; Full AMD retpoline, IBPB conditional, IBRS_FW, STIBP always-on, RSB filling
Vulnerability Srbds:             Not affected
Vulnerability Tsx async abort:   Not affected

==============================
Versions of relevant libraries
==============================
[pip3] conch-triton-kernels==1.2.1
[pip3] numpy==2.1.3
[pip3] onnx==1.19.0
[pip3] onnx-ir==0.2.0
[pip3] onnxscript==0.6.2
[pip3] onnxslim==0.1.87
[pip3] pyzmq==27.1.0
[pip3] torch==2.9.1+git8907517
[pip3] torchaudio==2.9.0+eaa9e4e
[pip3] torchvision==0.24.1+d801a34
[pip3] transformers==4.57.6
[pip3] triton==3.4.0
[pip3] triton_kernels==1.0.0
[conda] Could not collect

==============================
         vLLM Info
==============================
ROCM Version                 : 7.0.51831-a3e329ad8
vLLM Version                 : 0.17.0
vLLM Build Flags:
  CUDA Archs: Not Set; ROCm: Disabled
GPU Topology:
  ============================ ROCm System Management Interface ============================
================================ Weight between two GPUs =================================
       GPU0         GPU1         GPU2         GPU3         GPU4         GPU5         GPU6         GPU7         
GPU0   0            15           15           15           15           15           15           15           
GPU1   15           0            15           15           15           15           15           15           
GPU2   15           15           0            15           15           15           15           15           
GPU3   15           15           15           0            15           15           15           15           
GPU4   15           15           15           15           0            15           15           15           
GPU5   15           15           15           15           15           0            15           15           
GPU6   15           15           15           15           15           15           0            15           
GPU7   15           15           15           15           15           15           15           0            

================================= Hops between two GPUs ==================================
       GPU0         GPU1         GPU2         GPU3         GPU4         GPU5         GPU6         GPU7         
GPU0   0            1            1            1            1            1            1            1            
GPU1   1            0            1            1            1            1            1            1            
GPU2   1            1            0            1            1            1            1            1            
GPU3   1            1            1            0            1            1            1            1            
GPU4   1            1            1            1            0            1            1            1            
GPU5   1            1            1            1            1            0            1            1            
GPU6   1            1            1            1            1            1            0            1            
GPU7   1            1            1            1            1            1            1            0            

=============================== Link Type between two GPUs ===============================
       GPU0         GPU1         GPU2         GPU3         GPU4         GPU5         GPU6         GPU7         
GPU0   0            XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         
GPU1   XGMI         0            XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         
GPU2   XGMI         XGMI         0            XGMI         XGMI         XGMI         XGMI         XGMI         
GPU3   XGMI         XGMI         XGMI         0            XGMI         XGMI         XGMI         XGMI         
GPU4   XGMI         XGMI         XGMI         XGMI         0            XGMI         XGMI         XGMI         
GPU5   XGMI         XGMI         XGMI         XGMI         XGMI         0            XGMI         XGMI         
GPU6   XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         0            XGMI         
GPU7   XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         0            

======================================= Numa Nodes =======================================
GPU[0]          : (Topology) Numa Node: 0
GPU[0]          : (Topology) Numa Affinity: 0
GPU[1]          : (Topology) Numa Node: 0
GPU[1]          : (Topology) Numa Affinity: 0
GPU[2]          : (Topology) Numa Node: 0
GPU[2]          : (Topology) Numa Affinity: 0
GPU[3]          : (Topology) Numa Node: 0
GPU[3]          : (Topology) Numa Affinity: 0
GPU[4]          : (Topology) Numa Node: 1
GPU[4]          : (Topology) Numa Affinity: 1
GPU[5]          : (Topology) Numa Node: 1
GPU[5]          : (Topology) Numa Affinity: 1
GPU[6]          : (Topology) Numa Node: 1
GPU[6]          : (Topology) Numa Affinity: 1
GPU[7]          : (Topology) Numa Node: 1
GPU[7]          : (Topology) Numa Affinity: 1
================================== End of ROCm SMI Log ===================================

==============================
     Environment Variables
==============================
NVIDIA_DISABLE_REQUIRE=1
PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151
LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
PYTORCH_NVML_BASED_CUDA_CHECK=1
TORCHINDUCTOR_COMPILE_THREADS=1

---

vllm serve Step-3.5-Flash-FP8 \
  --served-model-name step3p5-flash \
  --tensor-parallel-size 8 \
  --enable-expert-parallel \
  --max-num-seqs 512 \
  --disable-cascade-attn \
  --reasoning-parser step3p5 \
  --enable-auto-tool-choice \
  --tool-call-parser step3p5 \
  --trust-remote-code \
  --async-scheduling \
  --no-enable-prefix-caching \
  --compilation-config '{"cudagraph_mode": "PIECEWISE"}' \
  --attention-config.backend ROCM_AITER_UNIFIED_ATTN \
  --speculative_config '{"method": "step3p5_mtp", "num_speculative_tokens": 1}' \
  --quantization fp8

---

(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] WorkerProc hit an exception.
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] Traceback (most recent call last):
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/multiproc_executor.py", line 875, in worker_busy_loop
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     output = func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]              ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/worker_base.py", line 320, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.worker.initialize_from_config(kv_cache_config)  # type: ignore
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     return func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]            ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 481, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.model_runner.initialize_kv_cache(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 6071, in initialize_kv_cache
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.initialize_metadata_builders(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5557, in initialize_metadata_builders
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.drafter.initialize_attn_backend(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1596, in initialize_attn_backend
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.validate_same_kv_cache_group(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1570, in validate_same_kv_cache_group
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     len(
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] AssertionError: All drafting layers should belong to the same kv cache group
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] Traceback (most recent call last):
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/multiproc_executor.py", line 875, in worker_busy_loop
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     output = func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]              ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/worker_base.py", line 320, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.worker.initialize_from_config(kv_cache_config)  # type: ignore
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     return func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]            ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 481, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.model_runner.initialize_kv_cache(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 6071, in initialize_kv_cache
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.initialize_metadata_builders(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5557, in initialize_metadata_builders
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.drafter.initialize_attn_backend(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1596, in initialize_attn_backend
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.validate_same_kv_cache_group(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1570, in validate_same_kv_cache_group
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     len(
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] AssertionError: All drafting layers should belong to the same kv cache group
RAW_BUFFERClick to expand / collapse

Your current environment

<details> <summary>The output of <code>python collect_env.py</code></summary>
Collecting environment information...
==============================
        System Info
==============================
OS                           : Ubuntu 22.04.5 LTS (x86_64)
GCC version                  : (Ubuntu 11.4.0-1ubuntu1~22.04.2) 11.4.0
Clang version                : 20.0.0git (https://github.com/RadeonOpenCompute/llvm-project roc-7.0.0 25314 f4087f6b428f0e6f575ebac8a8a724dab123d06e)
CMake version                : version 3.31.10
Libc version                 : glibc-2.35

==============================
       PyTorch Info
==============================
PyTorch version              : 2.9.1+git8907517
Is debug build               : False
CUDA used to build PyTorch   : N/A
ROCM used to build PyTorch   : 7.0.51831-a3e329ad8

==============================
      Python Environment
==============================
Python version               : 3.12.12 (main, Oct 10 2025, 08:52:57) [GCC 11.4.0] (64-bit runtime)
Python platform              : Linux-5.4.119-19.0009.60-x86_64-with-glibc2.35

==============================
       CUDA / GPU Info
==============================
Is CUDA available            : True
CUDA runtime version         : Could not collect
CUDA_MODULE_LOADING set to   : 
GPU models and configuration :  (gfx942:sramecc+:xnack-)
Nvidia driver version        : Could not collect
cuDNN version                : Could not collect
HIP runtime version          : 7.0.51831
MIOpen runtime version       : 3.5.0
Is XNNPACK available         : True

==============================
          CPU Info
==============================
Architecture:                    x86_64
CPU op-mode(s):                  32-bit, 64-bit
Address sizes:                   52 bits physical, 57 bits virtual
Byte Order:                      Little Endian
CPU(s):                          384
On-line CPU(s) list:             0-383
Vendor ID:                       AuthenticAMD
BIOS Vendor ID:                  Advanced Micro Devices, Inc.
Model name:                      AMD EPYC 9K84 96-Core Processor
BIOS Model name:                 AMD EPYC 9K84 96-Core Processor                
CPU family:                      25
Model:                           17
Thread(s) per core:              2
Core(s) per socket:              96
Socket(s):                       2
Stepping:                        1
Frequency boost:                 enabled
CPU max MHz:                     2600.0000
CPU min MHz:                     1500.0000
BogoMIPS:                        5200.50
Flags:                           fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush mmx fxsr sse sse2 ht syscall nx mmxext fxsr_opt pdpe1gb rdtscp lm constant_tsc rep_good nopl nonstop_tsc cpuid extd_apicid aperfmperf pni pclmulqdq monitor ssse3 fma cx16 pcid sse4_1 sse4_2 x2apic movbe popcnt aes xsave avx f16c rdrand lahf_lm cmp_legacy svm extapic cr8_legacy abm sse4a misalignsse 3dnowprefetch osvw ibs skinit wdt tce topoext perfctr_core perfctr_nb bpext perfctr_llc mwaitx cpb cat_l3 cdp_l3 invpcid_single hw_pstate ssbd mba ibrs ibpb stibp vmmcall fsgsbase bmi1 avx2 smep bmi2 erms invpcid cqm rdt_a avx512f avx512dq rdseed adx smap avx512ifma clflushopt clwb avx512cd sha_ni avx512bw avx512vl xsaveopt xsavec xgetbv1 xsaves cqm_llc cqm_occup_llc cqm_mbm_total cqm_mbm_local avx512_bf16 clzero irperf xsaveerptr wbnoinvd arat npt lbrv svm_lock nrip_save tsc_scale vmcb_clean flushbyasid decodeassists pausefilter pfthreshold avic v_vmsave_vmload vgif avx512vbmi umip pku ospke avx512_vbmi2 gfni vaes vpclmulqdq avx512_vnni avx512_bitalg avx512_vpopcntdq la57 rdpid overflow_recov succor smca fsrm flush_l1d
Virtualization:                  AMD-V
L1d cache:                       6 MiB (192 instances)
L1i cache:                       6 MiB (192 instances)
L2 cache:                        192 MiB (192 instances)
L3 cache:                        768 MiB (24 instances)
NUMA node(s):                    2
NUMA node0 CPU(s):               0-95,192-287
NUMA node1 CPU(s):               96-191,288-383
Vulnerability Itlb multihit:     Not affected
Vulnerability L1tf:              Not affected
Vulnerability Mds:               Not affected
Vulnerability Meltdown:          Not affected
Vulnerability Spec store bypass: Mitigation; Speculative Store Bypass disabled via prctl and seccomp
Vulnerability Spectre v1:        Mitigation; usercopy/swapgs barriers and __user pointer sanitization
Vulnerability Spectre v2:        Mitigation; Full AMD retpoline, IBPB conditional, IBRS_FW, STIBP always-on, RSB filling
Vulnerability Srbds:             Not affected
Vulnerability Tsx async abort:   Not affected

==============================
Versions of relevant libraries
==============================
[pip3] conch-triton-kernels==1.2.1
[pip3] numpy==2.1.3
[pip3] onnx==1.19.0
[pip3] onnx-ir==0.2.0
[pip3] onnxscript==0.6.2
[pip3] onnxslim==0.1.87
[pip3] pyzmq==27.1.0
[pip3] torch==2.9.1+git8907517
[pip3] torchaudio==2.9.0+eaa9e4e
[pip3] torchvision==0.24.1+d801a34
[pip3] transformers==4.57.6
[pip3] triton==3.4.0
[pip3] triton_kernels==1.0.0
[conda] Could not collect

==============================
         vLLM Info
==============================
ROCM Version                 : 7.0.51831-a3e329ad8
vLLM Version                 : 0.17.0
vLLM Build Flags:
  CUDA Archs: Not Set; ROCm: Disabled
GPU Topology:
  ============================ ROCm System Management Interface ============================
================================ Weight between two GPUs =================================
       GPU0         GPU1         GPU2         GPU3         GPU4         GPU5         GPU6         GPU7         
GPU0   0            15           15           15           15           15           15           15           
GPU1   15           0            15           15           15           15           15           15           
GPU2   15           15           0            15           15           15           15           15           
GPU3   15           15           15           0            15           15           15           15           
GPU4   15           15           15           15           0            15           15           15           
GPU5   15           15           15           15           15           0            15           15           
GPU6   15           15           15           15           15           15           0            15           
GPU7   15           15           15           15           15           15           15           0            

================================= Hops between two GPUs ==================================
       GPU0         GPU1         GPU2         GPU3         GPU4         GPU5         GPU6         GPU7         
GPU0   0            1            1            1            1            1            1            1            
GPU1   1            0            1            1            1            1            1            1            
GPU2   1            1            0            1            1            1            1            1            
GPU3   1            1            1            0            1            1            1            1            
GPU4   1            1            1            1            0            1            1            1            
GPU5   1            1            1            1            1            0            1            1            
GPU6   1            1            1            1            1            1            0            1            
GPU7   1            1            1            1            1            1            1            0            

=============================== Link Type between two GPUs ===============================
       GPU0         GPU1         GPU2         GPU3         GPU4         GPU5         GPU6         GPU7         
GPU0   0            XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         
GPU1   XGMI         0            XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         
GPU2   XGMI         XGMI         0            XGMI         XGMI         XGMI         XGMI         XGMI         
GPU3   XGMI         XGMI         XGMI         0            XGMI         XGMI         XGMI         XGMI         
GPU4   XGMI         XGMI         XGMI         XGMI         0            XGMI         XGMI         XGMI         
GPU5   XGMI         XGMI         XGMI         XGMI         XGMI         0            XGMI         XGMI         
GPU6   XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         0            XGMI         
GPU7   XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         XGMI         0            

======================================= Numa Nodes =======================================
GPU[0]          : (Topology) Numa Node: 0
GPU[0]          : (Topology) Numa Affinity: 0
GPU[1]          : (Topology) Numa Node: 0
GPU[1]          : (Topology) Numa Affinity: 0
GPU[2]          : (Topology) Numa Node: 0
GPU[2]          : (Topology) Numa Affinity: 0
GPU[3]          : (Topology) Numa Node: 0
GPU[3]          : (Topology) Numa Affinity: 0
GPU[4]          : (Topology) Numa Node: 1
GPU[4]          : (Topology) Numa Affinity: 1
GPU[5]          : (Topology) Numa Node: 1
GPU[5]          : (Topology) Numa Affinity: 1
GPU[6]          : (Topology) Numa Node: 1
GPU[6]          : (Topology) Numa Affinity: 1
GPU[7]          : (Topology) Numa Node: 1
GPU[7]          : (Topology) Numa Affinity: 1
================================== End of ROCm SMI Log ===================================

==============================
     Environment Variables
==============================
NVIDIA_DISABLE_REQUIRE=1
PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151
LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
PYTORCH_NVML_BASED_CUDA_CHECK=1
TORCHINDUCTOR_COMPILE_THREADS=1
</details>

🐛 Describe the bug

vllm serve Step-3.5-Flash-FP8 \
  --served-model-name step3p5-flash \
  --tensor-parallel-size 8 \
  --enable-expert-parallel \
  --max-num-seqs 512 \
  --disable-cascade-attn \
  --reasoning-parser step3p5 \
  --enable-auto-tool-choice \
  --tool-call-parser step3p5 \
  --trust-remote-code \
  --async-scheduling \
  --no-enable-prefix-caching \
  --compilation-config '{"cudagraph_mode": "PIECEWISE"}' \
  --attention-config.backend ROCM_AITER_UNIFIED_ATTN \
  --speculative_config '{"method": "step3p5_mtp", "num_speculative_tokens": 1}' \
  --quantization fp8

then got the error

(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] WorkerProc hit an exception.
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] Traceback (most recent call last):
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/multiproc_executor.py", line 875, in worker_busy_loop
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     output = func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]              ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/worker_base.py", line 320, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.worker.initialize_from_config(kv_cache_config)  # type: ignore
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     return func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]            ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 481, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.model_runner.initialize_kv_cache(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 6071, in initialize_kv_cache
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.initialize_metadata_builders(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5557, in initialize_metadata_builders
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.drafter.initialize_attn_backend(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1596, in initialize_attn_backend
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.validate_same_kv_cache_group(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1570, in validate_same_kv_cache_group
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     len(
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] AssertionError: All drafting layers should belong to the same kv cache group
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] Traceback (most recent call last):
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/multiproc_executor.py", line 875, in worker_busy_loop
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     output = func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]              ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/worker_base.py", line 320, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.worker.initialize_from_config(kv_cache_config)  # type: ignore
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     return func(*args, **kwargs)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]            ^^^^^^^^^^^^^^^^^^^^^
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 481, in initialize_from_config
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.model_runner.initialize_kv_cache(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 6071, in initialize_kv_cache
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.initialize_metadata_builders(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5557, in initialize_metadata_builders
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.drafter.initialize_attn_backend(kv_cache_config, kernel_block_sizes)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1596, in initialize_attn_backend
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     self.validate_same_kv_cache_group(kv_cache_config)
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/spec_decode/eagle.py", line 1570, in validate_same_kv_cache_group
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880]     len(
(Worker pid=96931) (Worker_TP1_EP1 pid=96931) ERROR 03-30 03:13:32 [multiproc_executor.py:880] AssertionError: All drafting layers should belong to the same kv cache group

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

extent analysis

Fix Plan

The error message indicates that all drafting layers should belong to the same kv cache group. To fix this issue, we need to ensure that the kv_cache_config is properly set for all drafting layers.

Here are the steps to fix the issue:

  • Check the kv_cache_config in the initialize_from_config method of the gpu_model_runner.py file.
  • Verify that the kv_cache_config is the same for all drafting layers.
  • If the kv_cache_config is different for some drafting layers, update the configuration to use the same kv_cache_config for all layers.

Example code to update the kv_cache_config:

# In gpu_model_runner.py
def initialize_from_config(self, kv_cache_config):
    # ...
    self.drafter.initialize_attn_backend(kv_cache_config, kernel_block_sizes)
    # ...

# Update the kv_cache_config to use the same config for all drafting layers
kv_cache_config = {
    'kv_cache_group': 'same_group'
}

# Pass the updated kv_cache_config to the initialize_from_config method
self.initialize_from_config(kv_cache_config)

Verification

To verify that the fix worked, run the vllm serve command again with the updated kv_cache_config. If the error is resolved, the command should run without any issues.

Extra Tips

  • Make sure to check the documentation for any specific requirements for the kv_cache_config in your version of the library.
  • If you are still experiencing issues, try to isolate the problem by testing individual drafting layers with different kv_cache_config settings.

Vote matrix · Quick signals

Works
Did the solution work? Tap to confirm.
Easy Fix
Was it a quick fix?
Time Saver
Did it save you time?
Blocking
Was it severely blocking?
Common Issue
Are others likely hitting this too?
Flaky / Intermittent
Is it intermittent?
Verified / Reproducible
Can you reproduce it reliably?
Loading…

Still need to ship something?

×6

Another batch ranked right after the header list — different links, same matching logic.

Back to top recommendations

TRENDING