vllm - 💡(How to fix) Fix [Bug]: GPT OSS 120B performance regression with Triton 3.6 [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#37441Fetched 2026-04-08 00:58:38
View on GitHub
Comments
4
Participants
3
Timeline
9
Reactions
1
Author
Timeline (top)
commented ×4subscribed ×3labeled ×1mentioned ×1

Error Message

vllm serve {PATH_TO_MODEL} --host 0.0.0.0 --port 8000 --served-model-name oss --tensor-parallel-size 2 --uvicorn_log_level error --max_num_seqs 2 --trust-remote-code --gpu-memory-utilization 0.9 --max-model-len 131072 --enable-chunked-prefill --max-num-batched-tokens 8192 --speculative-config.method eagle --speculative-config.num_speculative_tokens 3 --max-cudagraph-capture-size 4096 --enable-prefix-caching --stream-interval 20 --speculative-config.model {PATH_TO_DRAFT_MODEL}

Fix Action

Fix / Workaround

Nvidia driver version : 570.195.03 cuDNN version : Could not collect HIP runtime version : N/A MIOpen runtime version : N/A Is XNNPACK available : True

      CPU Info

============================== Architecture: x86_64 CPU op-mode(s): 32-bit, 64-bit Address sizes: 46 bits physical, 48 bits virtual Byte Order: Little Endian CPU(s): 192 On-line CPU(s) list: 0-191 Vendor ID: GenuineIntel Model name: Intel(R) Xeon(R) Platinum 8488C CPU family: 6 Model: 143 Thread(s) per core: 2 Core(s) per socket: 48 Socket(s): 2 Stepping: 8 BogoMIPS: 4800.00 Flags: fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush mmx fxsr sse sse2 ss ht syscall nx pdpe1gb rdtscp lm constant_tsc arch_perfmon rep_good nopl xtopology nonstop_tsc cpuid aperfmperf tsc_known_freq pni pclmulqdq monitor ssse3 fma cx16 pdcm pcid sse4_1 sse4_2 x2apic movbe popcnt tsc_deadline_timer aes xsave avx f16c rdrand hypervisor lahf_lm abm 3dnowprefetch invpcid_single ssbd ibrs ibpb stibp ibrs_enhanced fsgsbase tsc_adjust bmi1 avx2 smep bmi2 erms invpcid avx512f avx512dq rdseed adx smap avx512ifma clflushopt clwb avx512cd sha_ni avx512bw avx512vl xsaveopt xsavec xgetbv1 xsaves avx512_bf16 wbnoinvd ida arat avx512vbmi umip pku ospke waitpkg avx512_vbmi2 gfni vaes vpclmulqdq avx512_vnni avx512_bitalg tme avx512_vpopcntdq rdpid cldemote movdiri movdir64b md_clear serialize flush_l1d arch_capabilities Hypervisor vendor: KVM Virtualization type: full L1d cache: 4.5 MiB (96 instances) L1i cache: 3 MiB (96 instances) L2 cache: 192 MiB (96 instances) L3 cache: 210 MiB (2 instances) NUMA node(s): 2 NUMA node0 CPU(s): 0-47,96-143 NUMA node1 CPU(s): 48-95,144-191 Vulnerability Gather data sampling: Not affected Vulnerability Indirect target selection: Not affected Vulnerability Itlb multihit: Not affected Vulnerability L1tf: Not affected Vulnerability Mds: Not affected Vulnerability Meltdown: Not affected Vulnerability Mmio stale data: Not affected Vulnerability Reg file data sampling: Not affected Vulnerability Retbleed: Not affected Vulnerability Spec rstack overflow: 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; Enhanced / Automatic IBRS, IBPB conditional, RSB filling, PBRSB-eIBRS SW sequence Vulnerability Srbds: Not affected Vulnerability Tsa: Not affected Vulnerability Tsx async abort: Not affected Vulnerability Vmscape: Not affected

Running GPT OSS 120B on H200 GPU become slower with upgrade to vLLM 0.17.0 compared to 0.16.0 with ~20% latency regression. After profiling execution with nsys, found that 0.17.1 uses new kernel _reduce in 0.17.1 (Triton 3.6) vs reduce_grouped in 0.16 (Triton 3.5), which are coming from Triton MoE kernels. The _reduce takes 6.4 times longer than reduce_grouped. Then we tried to revert back Triton 3.5 kernels while keeping vLLM 0.17.1 and updating gpt_oss_triton_kernels_moe.py to always use_legacy_triton_kernels = True (link), this recovered the previous GPT OSS performance fully.

Code Example

==============================
        System Info
==============================
OS                           : Ubuntu 22.04.5 LTS (x86_64)
GCC version                  : (Ubuntu 11.4.0-1ubuntu1~22.04.3) 11.4.0
Clang version                : Could not collect
CMake version                : version 3.22.1
Libc version                 : glibc-2.35

==============================
       PyTorch Info
==============================
PyTorch version              : 2.10.0+cu129
Is debug build               : False
CUDA used to build PyTorch   : 12.9
ROCM used to build PyTorch   : N/A

==============================
      Python Environment
==============================
Python version               : 3.12.13 (main, Mar  4 2026, 09:23:07) [GCC 11.4.0] (64-bit runtime)
Python platform              : Linux-5.10.245-245.983.amzn2.x86_64-x86_64-with-glibc2.35
==============================
       CUDA / GPU Info
==============================
Is CUDA available            : True
CUDA runtime version         : 12.9.86
CUDA_MODULE_LOADING set to   : 
GPU models and configuration : 
GPU 0: NVIDIA H200
GPU 1: NVIDIA H200
GPU 2: NVIDIA H200
GPU 3: NVIDIA H200
GPU 4: NVIDIA H200
GPU 5: NVIDIA H200
GPU 6: NVIDIA H200
GPU 7: NVIDIA H200

Nvidia driver version        : 570.195.03
cuDNN version                : Could not collect
HIP runtime version          : N/A
MIOpen runtime version       : N/A
Is XNNPACK available         : True
==============================
          CPU Info
==============================
Architecture:                            x86_64
CPU op-mode(s):                          32-bit, 64-bit
Address sizes:                           46 bits physical, 48 bits virtual
Byte Order:                              Little Endian
CPU(s):                                  192
On-line CPU(s) list:                     0-191
Vendor ID:                               GenuineIntel
Model name:                              Intel(R) Xeon(R) Platinum 8488C
CPU family:                              6
Model:                                   143
Thread(s) per core:                      2
Core(s) per socket:                      48
Socket(s):                               2
Stepping:                                8
BogoMIPS:                                4800.00
Flags:                                   fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush mmx fxsr sse sse2 ss ht syscall nx pdpe1gb rdtscp lm constant_tsc arch_perfmon rep_good nopl xtopology nonstop_tsc cpuid aperfmperf tsc_known_freq pni pclmulqdq monitor ssse3 fma cx16 pdcm pcid sse4_1 sse4_2 x2apic movbe popcnt tsc_deadline_timer aes xsave avx f16c rdrand hypervisor lahf_lm abm 3dnowprefetch invpcid_single ssbd ibrs ibpb stibp ibrs_enhanced fsgsbase tsc_adjust bmi1 avx2 smep bmi2 erms invpcid avx512f avx512dq rdseed adx smap avx512ifma clflushopt clwb avx512cd sha_ni avx512bw avx512vl xsaveopt xsavec xgetbv1 xsaves avx512_bf16 wbnoinvd ida arat avx512vbmi umip pku ospke waitpkg avx512_vbmi2 gfni vaes vpclmulqdq avx512_vnni avx512_bitalg tme avx512_vpopcntdq rdpid cldemote movdiri movdir64b md_clear serialize flush_l1d arch_capabilities
Hypervisor vendor:                       KVM
Virtualization type:                     full
L1d cache:                               4.5 MiB (96 instances)
L1i cache:                               3 MiB (96 instances)
L2 cache:                                192 MiB (96 instances)
L3 cache:                                210 MiB (2 instances)
NUMA node(s):                            2
NUMA node0 CPU(s):                       0-47,96-143
NUMA node1 CPU(s):                       48-95,144-191
Vulnerability Gather data sampling:      Not affected
Vulnerability Indirect target selection: Not affected
Vulnerability Itlb multihit:             Not affected
Vulnerability L1tf:                      Not affected
Vulnerability Mds:                       Not affected
Vulnerability Meltdown:                  Not affected
Vulnerability Mmio stale data:           Not affected
Vulnerability Reg file data sampling:    Not affected
Vulnerability Retbleed:                  Not affected
Vulnerability Spec rstack overflow:      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; Enhanced / Automatic IBRS, IBPB conditional, RSB filling, PBRSB-eIBRS SW sequence
Vulnerability Srbds:                     Not affected
Vulnerability Tsa:                       Not affected
Vulnerability Tsx async abort:           Not affected
Vulnerability Vmscape:                   Not affected

==============================
Versions of relevant libraries
==============================
[pip3] flashinfer-python==0.6.4
[pip3] numpy==2.2.6
[pip3] nvidia-cublas-cu12==12.9.1.4
[pip3] nvidia-cuda-cupti-cu12==12.9.79
[pip3] nvidia-cuda-nvrtc-cu12==12.9.86
[pip3] nvidia-cuda-runtime-cu12==12.9.79
[pip3] nvidia-cudnn-cu12==9.10.2.21
[pip3] nvidia-cudnn-frontend==1.18.0
[pip3] nvidia-cufft-cu12==11.4.1.4
[pip3] nvidia-cufile-cu12==1.14.1.1
[pip3] nvidia-curand-cu12==10.3.10.19
[pip3] nvidia-cusolver-cu12==11.7.5.82
[pip3] nvidia-cusparse-cu12==12.5.10.65
[pip3] nvidia-cusparselt-cu12==0.7.1
[pip3] nvidia-cutlass-dsl==4.4.1
[pip3] nvidia-cutlass-dsl-libs-base==4.4.1
[pip3] nvidia-ml-py==13.590.48
[pip3] nvidia-nccl-cu12==2.27.5
[pip3] nvidia-nvjitlink-cu12==12.9.86
[pip3] nvidia-nvshmem-cu12==3.4.5
[pip3] nvidia-nvtx-cu12==12.9.79
[pip3] pyzmq==27.1.0
[pip3] torch==2.10.0+cu129
[pip3] torch_c_dlpack_ext==0.1.5
[pip3] torchaudio==2.10.0+cu129
[pip3] torchvision==0.25.0+cu129
[pip3] transformers==4.57.6
[pip3] triton==3.6.0
[pip3] triton_kernels==1.0.0
[conda] Could not collect

==============================
         vLLM Info
==============================
ROCM Version                 : Could not collect
vLLM Version                 : 0.17.1
vLLM Build Flags:
  CUDA Archs: 7.0 7.5 8.0 8.9 9.0 10.0 12.0; ROCm: Disabled
GPU Topology:
        GPU0    GPU1    GPU2    GPU3    GPU4    GPU5    GPU6    GPU7    CPU Affinity    NUMA AffinityGPU NUMA ID
GPU0     X      NV18    NV18    NV18    NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU1    NV18     X      NV18    NV18    NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU2    NV18    NV18     X      NV18    NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU3    NV18    NV18    NV18     X      NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU4    NV18    NV18    NV18    NV18     X      NV18    NV18    NV18    48-95,144-191   1           N/A
GPU5    NV18    NV18    NV18    NV18    NV18     X      NV18    NV18    48-95,144-191   1           N/A
GPU6    NV18    NV18    NV18    NV18    NV18    NV18     X      NV18    48-95,144-191   1           N/A
GPU7    NV18    NV18    NV18    NV18    NV18    NV18    NV18     X      48-95,144-191   1           N/A

Legend:

  X    = Self
  SYS  = Connection traversing PCIe as well as the SMP interconnect between NUMA nodes (e.g., QPI/UPI)
  NODE = Connection traversing PCIe as well as the interconnect between PCIe Host Bridges within a NUMA node
  PHB  = Connection traversing PCIe as well as a PCIe Host Bridge (typically the CPU)
  PXB  = Connection traversing multiple PCIe bridges (without traversing the PCIe Host Bridge)
  PIX  = Connection traversing at most a single PCIe bridge
  NV#  = Connection traversing a bonded set of # NVLinks

---

vllm serve {PATH_TO_MODEL} --host 0.0.0.0 --port 8000 --served-model-name oss --tensor-parallel-size 2 --uvicorn_log_level error --max_num_seqs 2 --trust-remote-code --gpu-memory-utilization 0.9 --max-model-len 131072 --enable-chunked-prefill --max-num-batched-tokens 8192 --speculative-config.method eagle --speculative-config.num_speculative_tokens 3 --max-cudagraph-capture-size 4096 --enable-prefix-caching --stream-interval 20 --speculative-config.model {PATH_TO_DRAFT_MODEL}

---

vllm bench serve --port 8000 --save-result --save-detailed \
  --backend vllm \
  --served-model-name oss \
  --endpoint /v1/completions \
  --dataset-name custom \
  --dataset-path custom-bench.jsonl  \
  --extra-body '{"max_tokens":4096,"stop_token_ids":[200002]}' \
  --max-concurrency 1 \
  --temperature=0.0 \
  --top-p=0.95 \
  --top-k=1 \
  --result-dir "./log/"

---

============ Serving Benchmark Result ============                                                   
Successful requests:                     1000                                                        
Failed requests:                         0                                                           
Maximum request concurrency:             1                                                           
Benchmark duration (s):                  452.37                                                      
Total input tokens:                      27308698                                                    
Total generated tokens:                  124781                                                      
Request throughput (req/s):              2.21                                                        
Output token throughput (tok/s):         275.84                                                      
Peak output token throughput (tok/s):    26.00                                                       
Peak concurrent requests:                6.00                                                        
Total token throughput (tok/s):          60643.93                                                    
---------------Time to First Token----------------                                                   
Mean TTFT (ms):                          183.51                                                      
Median TTFT (ms):                        134.69                                                                       
P99 TTFT (ms):                           787.26                                                      
-----Time per Output Token (excl. 1st token)------                                                   
Mean TPOT (ms):                          2.10                                                        
Median TPOT (ms):                        1.97                                                                          
P99 TPOT (ms):                           5.71                                                        
---------------Inter-token Latency---------------- 
Mean ITL (ms):                           42.38       
Median ITL (ms):                         41.58                                                       
P99 ITL (ms):                            118.22     
---------------Speculative Decoding--------------- 
Acceptance rate (%):                     59.08      
Acceptance length:                       2.77       
Drafts:                                  44838      
Draft tokens:                            134514     
Accepted tokens:                         79476      
Per-position acceptance (%):
  Position 0:                            75.23      
  Position 1:                            57.10      
  Position 2:                            44.92      
==================================================

---

============ Serving Benchmark Result ============                                                   
Successful requests:                     1000                                                        
Failed requests:                         0                                                           
Maximum request concurrency:             1                                                           
Benchmark duration (s):                  452.37                                                      
Total input tokens:                      27308698                                                    
Total generated tokens:                  124781                                                      
Request throughput (req/s):              1.92                                                       
Output token throughput (tok/s):         231.22                                                      
Peak output token throughput (tok/s):    26.00                                                       
Peak concurrent requests:                6.00                                                        
Total token throughput (tok/s):          48,515.14                                             
---------------Time to First Token----------------                                                   
Mean TTFT (ms):                          197.58                                                      
Median TTFT (ms):                        164.29                                                                       
P99 TTFT (ms):                           878.21                                                      
-----Time per Output Token (excl. 1st token)------                                                   
Mean TPOT (ms):                          2.54                                                        
Median TPOT (ms):                        2.46                                                                          
P99 TPOT (ms):                           6.89                                                        
---------------Inter-token Latency---------------- 
Mean ITL (ms):                           39.41       
Median ITL (ms):                         37.22                                                       
P99 ITL (ms):                            101.42     
---------------Speculative Decoding--------------- 
Acceptance rate (%):                     59.08      
Acceptance length:                       2.77       
Drafts:                                  44838      
Draft tokens:                            134514     
Accepted tokens:                         79476      
Per-position acceptance (%):
  Position 0:                            75.23      
  Position 1:                            57.10      
  Position 2:                            44.92      
==================================================

---

============ Serving Benchmark Result ============                                                   
Successful requests:                     1000                                                        
Failed requests:                         0                                                           
Maximum request concurrency:             1                                                           
Benchmark duration (s):                  452.37                                                      
Total input tokens:                      27308698                                                    
Total generated tokens:                  124781                                                      
Request throughput (req/s):              2.21                                                        
Output token throughput (tok/s):         275.84                                                      
Peak output token throughput (tok/s):    26.00                                                       
Peak concurrent requests:                6.00                                                        
Total token throughput (tok/s):          60643.93                                                    
---------------Time to First Token----------------                                                   
Mean TTFT (ms):                          181.34                                                      
Median TTFT (ms):                        137.21                                                                       
P99 TTFT (ms):                           768.13                                                      
-----Time per Output Token (excl. 1st token)------                                                   
Mean TPOT (ms):                          2.09                                                        
Median TPOT (ms):                        1.98                                                                          
P99 TPOT (ms):                           5.71                                                        
---------------Inter-token Latency---------------- 
Mean ITL (ms):                           42.38       
Median ITL (ms):                         41.58                                                       
P99 ITL (ms):                            118.22     
---------------Speculative Decoding--------------- 
Acceptance rate (%):                     59.08      
Acceptance length:                       2.77       
Drafts:                                  44838      
Draft tokens:                            134514     
Accepted tokens:                         79476      
Per-position acceptance (%):
  Position 0:                            75.23      
  Position 1:                            57.10      
  Position 2:                            44.92      
==================================================
RAW_BUFFERClick to expand / collapse

Your current environment

<details> <summary>The output of <code>python collect_env.py</code></summary>
==============================
        System Info
==============================
OS                           : Ubuntu 22.04.5 LTS (x86_64)
GCC version                  : (Ubuntu 11.4.0-1ubuntu1~22.04.3) 11.4.0
Clang version                : Could not collect
CMake version                : version 3.22.1
Libc version                 : glibc-2.35

==============================
       PyTorch Info
==============================
PyTorch version              : 2.10.0+cu129
Is debug build               : False
CUDA used to build PyTorch   : 12.9
ROCM used to build PyTorch   : N/A

==============================
      Python Environment
==============================
Python version               : 3.12.13 (main, Mar  4 2026, 09:23:07) [GCC 11.4.0] (64-bit runtime)
Python platform              : Linux-5.10.245-245.983.amzn2.x86_64-x86_64-with-glibc2.35
==============================
       CUDA / GPU Info
==============================
Is CUDA available            : True
CUDA runtime version         : 12.9.86
CUDA_MODULE_LOADING set to   : 
GPU models and configuration : 
GPU 0: NVIDIA H200
GPU 1: NVIDIA H200
GPU 2: NVIDIA H200
GPU 3: NVIDIA H200
GPU 4: NVIDIA H200
GPU 5: NVIDIA H200
GPU 6: NVIDIA H200
GPU 7: NVIDIA H200

Nvidia driver version        : 570.195.03
cuDNN version                : Could not collect
HIP runtime version          : N/A
MIOpen runtime version       : N/A
Is XNNPACK available         : True
==============================
          CPU Info
==============================
Architecture:                            x86_64
CPU op-mode(s):                          32-bit, 64-bit
Address sizes:                           46 bits physical, 48 bits virtual
Byte Order:                              Little Endian
CPU(s):                                  192
On-line CPU(s) list:                     0-191
Vendor ID:                               GenuineIntel
Model name:                              Intel(R) Xeon(R) Platinum 8488C
CPU family:                              6
Model:                                   143
Thread(s) per core:                      2
Core(s) per socket:                      48
Socket(s):                               2
Stepping:                                8
BogoMIPS:                                4800.00
Flags:                                   fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush mmx fxsr sse sse2 ss ht syscall nx pdpe1gb rdtscp lm constant_tsc arch_perfmon rep_good nopl xtopology nonstop_tsc cpuid aperfmperf tsc_known_freq pni pclmulqdq monitor ssse3 fma cx16 pdcm pcid sse4_1 sse4_2 x2apic movbe popcnt tsc_deadline_timer aes xsave avx f16c rdrand hypervisor lahf_lm abm 3dnowprefetch invpcid_single ssbd ibrs ibpb stibp ibrs_enhanced fsgsbase tsc_adjust bmi1 avx2 smep bmi2 erms invpcid avx512f avx512dq rdseed adx smap avx512ifma clflushopt clwb avx512cd sha_ni avx512bw avx512vl xsaveopt xsavec xgetbv1 xsaves avx512_bf16 wbnoinvd ida arat avx512vbmi umip pku ospke waitpkg avx512_vbmi2 gfni vaes vpclmulqdq avx512_vnni avx512_bitalg tme avx512_vpopcntdq rdpid cldemote movdiri movdir64b md_clear serialize flush_l1d arch_capabilities
Hypervisor vendor:                       KVM
Virtualization type:                     full
L1d cache:                               4.5 MiB (96 instances)
L1i cache:                               3 MiB (96 instances)
L2 cache:                                192 MiB (96 instances)
L3 cache:                                210 MiB (2 instances)
NUMA node(s):                            2
NUMA node0 CPU(s):                       0-47,96-143
NUMA node1 CPU(s):                       48-95,144-191
Vulnerability Gather data sampling:      Not affected
Vulnerability Indirect target selection: Not affected
Vulnerability Itlb multihit:             Not affected
Vulnerability L1tf:                      Not affected
Vulnerability Mds:                       Not affected
Vulnerability Meltdown:                  Not affected
Vulnerability Mmio stale data:           Not affected
Vulnerability Reg file data sampling:    Not affected
Vulnerability Retbleed:                  Not affected
Vulnerability Spec rstack overflow:      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; Enhanced / Automatic IBRS, IBPB conditional, RSB filling, PBRSB-eIBRS SW sequence
Vulnerability Srbds:                     Not affected
Vulnerability Tsa:                       Not affected
Vulnerability Tsx async abort:           Not affected
Vulnerability Vmscape:                   Not affected

==============================
Versions of relevant libraries
==============================
[pip3] flashinfer-python==0.6.4
[pip3] numpy==2.2.6
[pip3] nvidia-cublas-cu12==12.9.1.4
[pip3] nvidia-cuda-cupti-cu12==12.9.79
[pip3] nvidia-cuda-nvrtc-cu12==12.9.86
[pip3] nvidia-cuda-runtime-cu12==12.9.79
[pip3] nvidia-cudnn-cu12==9.10.2.21
[pip3] nvidia-cudnn-frontend==1.18.0
[pip3] nvidia-cufft-cu12==11.4.1.4
[pip3] nvidia-cufile-cu12==1.14.1.1
[pip3] nvidia-curand-cu12==10.3.10.19
[pip3] nvidia-cusolver-cu12==11.7.5.82
[pip3] nvidia-cusparse-cu12==12.5.10.65
[pip3] nvidia-cusparselt-cu12==0.7.1
[pip3] nvidia-cutlass-dsl==4.4.1
[pip3] nvidia-cutlass-dsl-libs-base==4.4.1
[pip3] nvidia-ml-py==13.590.48
[pip3] nvidia-nccl-cu12==2.27.5
[pip3] nvidia-nvjitlink-cu12==12.9.86
[pip3] nvidia-nvshmem-cu12==3.4.5
[pip3] nvidia-nvtx-cu12==12.9.79
[pip3] pyzmq==27.1.0
[pip3] torch==2.10.0+cu129
[pip3] torch_c_dlpack_ext==0.1.5
[pip3] torchaudio==2.10.0+cu129
[pip3] torchvision==0.25.0+cu129
[pip3] transformers==4.57.6
[pip3] triton==3.6.0
[pip3] triton_kernels==1.0.0
[conda] Could not collect

==============================
         vLLM Info
==============================
ROCM Version                 : Could not collect
vLLM Version                 : 0.17.1
vLLM Build Flags:
  CUDA Archs: 7.0 7.5 8.0 8.9 9.0 10.0 12.0; ROCm: Disabled
GPU Topology:
        GPU0    GPU1    GPU2    GPU3    GPU4    GPU5    GPU6    GPU7    CPU Affinity    NUMA AffinityGPU NUMA ID
GPU0     X      NV18    NV18    NV18    NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU1    NV18     X      NV18    NV18    NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU2    NV18    NV18     X      NV18    NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU3    NV18    NV18    NV18     X      NV18    NV18    NV18    NV18    0-47,96-143     0           N/A
GPU4    NV18    NV18    NV18    NV18     X      NV18    NV18    NV18    48-95,144-191   1           N/A
GPU5    NV18    NV18    NV18    NV18    NV18     X      NV18    NV18    48-95,144-191   1           N/A
GPU6    NV18    NV18    NV18    NV18    NV18    NV18     X      NV18    48-95,144-191   1           N/A
GPU7    NV18    NV18    NV18    NV18    NV18    NV18    NV18     X      48-95,144-191   1           N/A

Legend:

  X    = Self
  SYS  = Connection traversing PCIe as well as the SMP interconnect between NUMA nodes (e.g., QPI/UPI)
  NODE = Connection traversing PCIe as well as the interconnect between PCIe Host Bridges within a NUMA node
  PHB  = Connection traversing PCIe as well as a PCIe Host Bridge (typically the CPU)
  PXB  = Connection traversing multiple PCIe bridges (without traversing the PCIe Host Bridge)
  PIX  = Connection traversing at most a single PCIe bridge
  NV#  = Connection traversing a bonded set of # NVLinks
</details>

🐛 Describe the bug

Running GPT OSS 120B on H200 GPU become slower with upgrade to vLLM 0.17.0 compared to 0.16.0 with ~20% latency regression. After profiling execution with nsys, found that 0.17.1 uses new kernel _reduce in 0.17.1 (Triton 3.6) vs reduce_grouped in 0.16 (Triton 3.5), which are coming from Triton MoE kernels. The _reduce takes 6.4 times longer than reduce_grouped. Then we tried to revert back Triton 3.5 kernels while keeping vLLM 0.17.1 and updating gpt_oss_triton_kernels_moe.py to always use_legacy_triton_kernels = True (link), this recovered the previous GPT OSS performance fully.

Running vLLM with:

vllm serve {PATH_TO_MODEL} --host 0.0.0.0 --port 8000 --served-model-name oss --tensor-parallel-size 2 --uvicorn_log_level error --max_num_seqs 2 --trust-remote-code --gpu-memory-utilization 0.9 --max-model-len 131072 --enable-chunked-prefill --max-num-batched-tokens 8192 --speculative-config.method eagle --speculative-config.num_speculative_tokens 3 --max-cudagraph-capture-size 4096 --enable-prefix-caching --stream-interval 20 --speculative-config.model {PATH_TO_DRAFT_MODEL}

Running vLLM Benchmark with:

vllm bench serve --port 8000 --save-result --save-detailed \
  --backend vllm \
  --served-model-name oss \
  --endpoint /v1/completions \
  --dataset-name custom \
  --dataset-path custom-bench.jsonl  \
  --extra-body '{"max_tokens":4096,"stop_token_ids":[200002]}' \
  --max-concurrency 1 \
  --temperature=0.0 \
  --top-p=0.95 \
  --top-k=1 \
  --result-dir "./log/"

Benchmarking results for vLLM 0.16.0 with Triton 3.5:

============ Serving Benchmark Result ============                                                   
Successful requests:                     1000                                                        
Failed requests:                         0                                                           
Maximum request concurrency:             1                                                           
Benchmark duration (s):                  452.37                                                      
Total input tokens:                      27308698                                                    
Total generated tokens:                  124781                                                      
Request throughput (req/s):              2.21                                                        
Output token throughput (tok/s):         275.84                                                      
Peak output token throughput (tok/s):    26.00                                                       
Peak concurrent requests:                6.00                                                        
Total token throughput (tok/s):          60643.93                                                    
---------------Time to First Token----------------                                                   
Mean TTFT (ms):                          183.51                                                      
Median TTFT (ms):                        134.69                                                                       
P99 TTFT (ms):                           787.26                                                      
-----Time per Output Token (excl. 1st token)------                                                   
Mean TPOT (ms):                          2.10                                                        
Median TPOT (ms):                        1.97                                                                          
P99 TPOT (ms):                           5.71                                                        
---------------Inter-token Latency---------------- 
Mean ITL (ms):                           42.38       
Median ITL (ms):                         41.58                                                       
P99 ITL (ms):                            118.22     
---------------Speculative Decoding--------------- 
Acceptance rate (%):                     59.08      
Acceptance length:                       2.77       
Drafts:                                  44838      
Draft tokens:                            134514     
Accepted tokens:                         79476      
Per-position acceptance (%):
  Position 0:                            75.23      
  Position 1:                            57.10      
  Position 2:                            44.92      
==================================================

Benchmarking results for vLLM 0.17.1 with Triton 3.6:

============ Serving Benchmark Result ============                                                   
Successful requests:                     1000                                                        
Failed requests:                         0                                                           
Maximum request concurrency:             1                                                           
Benchmark duration (s):                  452.37                                                      
Total input tokens:                      27308698                                                    
Total generated tokens:                  124781                                                      
Request throughput (req/s):              1.92                                                       
Output token throughput (tok/s):         231.22                                                      
Peak output token throughput (tok/s):    26.00                                                       
Peak concurrent requests:                6.00                                                        
Total token throughput (tok/s):          48,515.14                                             
---------------Time to First Token----------------                                                   
Mean TTFT (ms):                          197.58                                                      
Median TTFT (ms):                        164.29                                                                       
P99 TTFT (ms):                           878.21                                                      
-----Time per Output Token (excl. 1st token)------                                                   
Mean TPOT (ms):                          2.54                                                        
Median TPOT (ms):                        2.46                                                                          
P99 TPOT (ms):                           6.89                                                        
---------------Inter-token Latency---------------- 
Mean ITL (ms):                           39.41       
Median ITL (ms):                         37.22                                                       
P99 ITL (ms):                            101.42     
---------------Speculative Decoding--------------- 
Acceptance rate (%):                     59.08      
Acceptance length:                       2.77       
Drafts:                                  44838      
Draft tokens:                            134514     
Accepted tokens:                         79476      
Per-position acceptance (%):
  Position 0:                            75.23      
  Position 1:                            57.10      
  Position 2:                            44.92      
==================================================

Benchmarking results for vLLM 0.17.1 with reverted Triton 3.5:

============ Serving Benchmark Result ============                                                   
Successful requests:                     1000                                                        
Failed requests:                         0                                                           
Maximum request concurrency:             1                                                           
Benchmark duration (s):                  452.37                                                      
Total input tokens:                      27308698                                                    
Total generated tokens:                  124781                                                      
Request throughput (req/s):              2.21                                                        
Output token throughput (tok/s):         275.84                                                      
Peak output token throughput (tok/s):    26.00                                                       
Peak concurrent requests:                6.00                                                        
Total token throughput (tok/s):          60643.93                                                    
---------------Time to First Token----------------                                                   
Mean TTFT (ms):                          181.34                                                      
Median TTFT (ms):                        137.21                                                                       
P99 TTFT (ms):                           768.13                                                      
-----Time per Output Token (excl. 1st token)------                                                   
Mean TPOT (ms):                          2.09                                                        
Median TPOT (ms):                        1.98                                                                          
P99 TPOT (ms):                           5.71                                                        
---------------Inter-token Latency---------------- 
Mean ITL (ms):                           42.38       
Median ITL (ms):                         41.58                                                       
P99 ITL (ms):                            118.22     
---------------Speculative Decoding--------------- 
Acceptance rate (%):                     59.08      
Acceptance length:                       2.77       
Drafts:                                  44838      
Draft tokens:                            134514     
Accepted tokens:                         79476      
Per-position acceptance (%):
  Position 0:                            75.23      
  Position 1:                            57.10      
  Position 2:                            44.92      
==================================================

Profiler kernel for vLLM 0.16.0 with Triton 3.5: <img width="913" height="504" alt="Image" src="https://github.com/user-attachments/assets/d6b9627f-6f53-45a5-8836-d600ddccef3a" />

Profiler kernel for vLLM 0.17.1 with Triton 3.6: <img width="910" height="481" alt="Image" src="https://github.com/user-attachments/assets/0e12ecae-68f5-4db4-addb-3d3ba65ef696" />

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

To fix the ~20% latency regression issue with vLLM 0.17.1, we need to revert back to Triton 3.5 kernels while keeping vLLM 0.17.1.

Here are the steps:

  • Update gpt_oss_triton_kernels_moe.py to always use_legacy_triton_kernels = True.
  • Revert Triton to version 3.5.

Example code changes:

# In gpt_oss_triton_kernels_moe.py
use_legacy_triton_kernels = True

You can also achieve this by using the --use-legacy-triton-kernels flag when running vLLM.

Verification

To verify that the fix worked, run the benchmarking tests again with the updated configuration. The results should be similar to those of vLLM 0.16.0 with Triton 3.5.

Extra Tips

  • Make sure to update the gpt_oss_triton_kernels_moe.py file correctly to avoid any errors.
  • If you are using a virtual environment, ensure that the correct version of Triton is installed.
  • You can also try to profile the execution with nsys to see if the _reduce kernel is still taking longer than reduce_grouped.

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