vllm-cpu-avx512vnni 0.10.2.post2__cp312-cp312-manylinux_2_17_x86_64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of vllm-cpu-avx512vnni might be problematic. Click here for more details.

Files changed (1395) hide show
  1. vllm/_C.abi3.so +0 -0
  2. vllm/__init__.py +220 -0
  3. vllm/_bc_linter.py +59 -0
  4. vllm/_custom_ops.py +2022 -0
  5. vllm/_ipex_ops.py +404 -0
  6. vllm/_version.py +34 -0
  7. vllm/adapter_commons/__init__.py +0 -0
  8. vllm/adapter_commons/layers.py +16 -0
  9. vllm/adapter_commons/models.py +106 -0
  10. vllm/adapter_commons/request.py +26 -0
  11. vllm/adapter_commons/utils.py +93 -0
  12. vllm/adapter_commons/worker_manager.py +39 -0
  13. vllm/assets/__init__.py +0 -0
  14. vllm/assets/audio.py +45 -0
  15. vllm/assets/base.py +41 -0
  16. vllm/assets/image.py +50 -0
  17. vllm/assets/video.py +138 -0
  18. vllm/attention/__init__.py +19 -0
  19. vllm/attention/backends/__init__.py +0 -0
  20. vllm/attention/backends/abstract.py +348 -0
  21. vllm/attention/backends/differential_flash_attn.py +935 -0
  22. vllm/attention/backends/dual_chunk_flash_attn.py +1499 -0
  23. vllm/attention/backends/flash_attn.py +933 -0
  24. vllm/attention/backends/flashmla.py +238 -0
  25. vllm/attention/backends/mla/__init__.py +0 -0
  26. vllm/attention/backends/mla/common.py +1310 -0
  27. vllm/attention/backends/placeholder_attn.py +340 -0
  28. vllm/attention/backends/rocm_aiter_mla.py +410 -0
  29. vllm/attention/backends/rocm_flash_attn.py +953 -0
  30. vllm/attention/backends/triton_mla.py +111 -0
  31. vllm/attention/backends/utils.py +610 -0
  32. vllm/attention/backends/xformers.py +805 -0
  33. vllm/attention/layer.py +552 -0
  34. vllm/attention/layers/__init__.py +0 -0
  35. vllm/attention/layers/chunked_local_attention.py +91 -0
  36. vllm/attention/layers/cross_attention.py +159 -0
  37. vllm/attention/layers/encoder_only_attention.py +86 -0
  38. vllm/attention/ops/__init__.py +0 -0
  39. vllm/attention/ops/chunked_prefill_paged_decode.py +405 -0
  40. vllm/attention/ops/common.py +139 -0
  41. vllm/attention/ops/flashmla.py +123 -0
  42. vllm/attention/ops/merge_attn_states.py +43 -0
  43. vllm/attention/ops/paged_attn.py +261 -0
  44. vllm/attention/ops/pallas_kv_cache_update.py +124 -0
  45. vllm/attention/ops/prefix_prefill.py +928 -0
  46. vllm/attention/ops/rocm_aiter_mla.py +104 -0
  47. vllm/attention/ops/rocm_aiter_paged_attn.py +102 -0
  48. vllm/attention/ops/triton_decode_attention.py +676 -0
  49. vllm/attention/ops/triton_flash_attention.py +984 -0
  50. vllm/attention/ops/triton_merge_attn_states.py +97 -0
  51. vllm/attention/ops/triton_unified_attention.py +854 -0
  52. vllm/attention/selector.py +243 -0
  53. vllm/attention/utils/__init__.py +0 -0
  54. vllm/attention/utils/fa_utils.py +85 -0
  55. vllm/attention/utils/kv_sharing_utils.py +33 -0
  56. vllm/beam_search.py +87 -0
  57. vllm/benchmarks/__init__.py +0 -0
  58. vllm/benchmarks/datasets.py +2651 -0
  59. vllm/benchmarks/latency.py +170 -0
  60. vllm/benchmarks/lib/__init__.py +3 -0
  61. vllm/benchmarks/lib/endpoint_request_func.py +510 -0
  62. vllm/benchmarks/lib/ready_checker.py +72 -0
  63. vllm/benchmarks/lib/utils.py +80 -0
  64. vllm/benchmarks/serve.py +1247 -0
  65. vllm/benchmarks/throughput.py +696 -0
  66. vllm/collect_env.py +823 -0
  67. vllm/compilation/__init__.py +0 -0
  68. vllm/compilation/activation_quant_fusion.py +193 -0
  69. vllm/compilation/backends.py +641 -0
  70. vllm/compilation/base_static_graph.py +51 -0
  71. vllm/compilation/collective_fusion.py +1190 -0
  72. vllm/compilation/compiler_interface.py +572 -0
  73. vllm/compilation/counter.py +47 -0
  74. vllm/compilation/cuda_graph.py +193 -0
  75. vllm/compilation/cuda_piecewise_backend.py +117 -0
  76. vllm/compilation/decorators.py +316 -0
  77. vllm/compilation/fix_functionalization.py +208 -0
  78. vllm/compilation/fusion.py +600 -0
  79. vllm/compilation/fusion_attn.py +303 -0
  80. vllm/compilation/fx_utils.py +84 -0
  81. vllm/compilation/inductor_pass.py +136 -0
  82. vllm/compilation/monitor.py +57 -0
  83. vllm/compilation/multi_output_match.py +109 -0
  84. vllm/compilation/noop_elimination.py +165 -0
  85. vllm/compilation/pass_manager.py +88 -0
  86. vllm/compilation/sequence_parallelism.py +484 -0
  87. vllm/compilation/torch25_custom_graph_pass.py +42 -0
  88. vllm/compilation/vllm_inductor_pass.py +50 -0
  89. vllm/compilation/wrapper.py +138 -0
  90. vllm/config/__init__.py +3921 -0
  91. vllm/config/cache.py +214 -0
  92. vllm/config/compilation.py +580 -0
  93. vllm/config/kv_events.py +50 -0
  94. vllm/config/kv_transfer.py +111 -0
  95. vllm/config/load.py +113 -0
  96. vllm/config/lora.py +132 -0
  97. vllm/config/parallel.py +446 -0
  98. vllm/config/scheduler.py +304 -0
  99. vllm/config/utils.py +29 -0
  100. vllm/connections.py +174 -0
  101. vllm/core/__init__.py +0 -0
  102. vllm/core/block/__init__.py +0 -0
  103. vllm/core/block/block_table.py +399 -0
  104. vllm/core/block/common.py +371 -0
  105. vllm/core/block/cpu_gpu_block_allocator.py +439 -0
  106. vllm/core/block/interfaces.py +319 -0
  107. vllm/core/block/naive_block.py +466 -0
  108. vllm/core/block/prefix_caching_block.py +1135 -0
  109. vllm/core/block/utils.py +28 -0
  110. vllm/core/block_manager.py +523 -0
  111. vllm/core/evictor.py +157 -0
  112. vllm/core/interfaces.py +139 -0
  113. vllm/core/placeholder_block_space_manager.py +103 -0
  114. vllm/core/scheduler.py +2028 -0
  115. vllm/device_allocator/__init__.py +0 -0
  116. vllm/device_allocator/cumem.py +286 -0
  117. vllm/distributed/__init__.py +6 -0
  118. vllm/distributed/communication_op.py +41 -0
  119. vllm/distributed/device_communicators/__init__.py +0 -0
  120. vllm/distributed/device_communicators/all2all.py +259 -0
  121. vllm/distributed/device_communicators/all_reduce_utils.py +292 -0
  122. vllm/distributed/device_communicators/base_device_communicator.py +277 -0
  123. vllm/distributed/device_communicators/cpu_communicator.py +201 -0
  124. vllm/distributed/device_communicators/cuda_communicator.py +294 -0
  125. vllm/distributed/device_communicators/cuda_wrapper.py +180 -0
  126. vllm/distributed/device_communicators/custom_all_reduce.py +311 -0
  127. vllm/distributed/device_communicators/pynccl.py +290 -0
  128. vllm/distributed/device_communicators/pynccl_wrapper.py +382 -0
  129. vllm/distributed/device_communicators/quick_all_reduce.py +278 -0
  130. vllm/distributed/device_communicators/ray_communicator.py +258 -0
  131. vllm/distributed/device_communicators/shm_broadcast.py +585 -0
  132. vllm/distributed/device_communicators/symm_mem.py +136 -0
  133. vllm/distributed/device_communicators/tpu_communicator.py +102 -0
  134. vllm/distributed/device_communicators/xpu_communicator.py +69 -0
  135. vllm/distributed/eplb/__init__.py +8 -0
  136. vllm/distributed/eplb/eplb_state.py +619 -0
  137. vllm/distributed/eplb/rebalance_algo.py +234 -0
  138. vllm/distributed/eplb/rebalance_execute.py +424 -0
  139. vllm/distributed/kv_events.py +362 -0
  140. vllm/distributed/kv_transfer/README.md +29 -0
  141. vllm/distributed/kv_transfer/__init__.py +13 -0
  142. vllm/distributed/kv_transfer/disagg_prefill_workflow.jpg +0 -0
  143. vllm/distributed/kv_transfer/kv_connector/__init__.py +0 -0
  144. vllm/distributed/kv_transfer/kv_connector/base.py +10 -0
  145. vllm/distributed/kv_transfer/kv_connector/factory.py +108 -0
  146. vllm/distributed/kv_transfer/kv_connector/utils.py +246 -0
  147. vllm/distributed/kv_transfer/kv_connector/v1/__init__.py +6 -0
  148. vllm/distributed/kv_transfer/kv_connector/v1/base.py +356 -0
  149. vllm/distributed/kv_transfer/kv_connector/v1/lmcache_connector.py +167 -0
  150. vllm/distributed/kv_transfer/kv_connector/v1/multi_connector.py +266 -0
  151. vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py +1319 -0
  152. vllm/distributed/kv_transfer/kv_connector/v1/p2p/__init__.py +0 -0
  153. vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_connector.py +484 -0
  154. vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_engine.py +542 -0
  155. vllm/distributed/kv_transfer/kv_connector/v1/p2p/tensor_memory_pool.py +266 -0
  156. vllm/distributed/kv_transfer/kv_connector/v1/shared_storage_connector.py +414 -0
  157. vllm/distributed/kv_transfer/kv_lookup_buffer/__init__.py +0 -0
  158. vllm/distributed/kv_transfer/kv_lookup_buffer/base.py +175 -0
  159. vllm/distributed/kv_transfer/kv_lookup_buffer/mooncake_store.py +161 -0
  160. vllm/distributed/kv_transfer/kv_lookup_buffer/simple_buffer.py +237 -0
  161. vllm/distributed/kv_transfer/kv_pipe/__init__.py +0 -0
  162. vllm/distributed/kv_transfer/kv_pipe/base.py +67 -0
  163. vllm/distributed/kv_transfer/kv_pipe/mooncake_pipe.py +290 -0
  164. vllm/distributed/kv_transfer/kv_pipe/pynccl_pipe.py +280 -0
  165. vllm/distributed/kv_transfer/kv_transfer_state.py +73 -0
  166. vllm/distributed/parallel_state.py +1489 -0
  167. vllm/distributed/tpu_distributed_utils.py +178 -0
  168. vllm/distributed/utils.py +536 -0
  169. vllm/engine/__init__.py +0 -0
  170. vllm/engine/arg_utils.py +1857 -0
  171. vllm/engine/async_llm_engine.py +1044 -0
  172. vllm/engine/async_timeout.py +173 -0
  173. vllm/engine/llm_engine.py +1849 -0
  174. vllm/engine/metrics.py +577 -0
  175. vllm/engine/metrics_types.py +84 -0
  176. vllm/engine/multiprocessing/__init__.py +145 -0
  177. vllm/engine/multiprocessing/client.py +643 -0
  178. vllm/engine/multiprocessing/engine.py +470 -0
  179. vllm/engine/output_processor/__init__.py +0 -0
  180. vllm/engine/output_processor/interfaces.py +61 -0
  181. vllm/engine/output_processor/single_step.py +145 -0
  182. vllm/engine/output_processor/stop_checker.py +131 -0
  183. vllm/engine/output_processor/util.py +28 -0
  184. vllm/engine/protocol.py +343 -0
  185. vllm/entrypoints/__init__.py +0 -0
  186. vllm/entrypoints/api_server.py +178 -0
  187. vllm/entrypoints/chat_utils.py +1535 -0
  188. vllm/entrypoints/cli/__init__.py +12 -0
  189. vllm/entrypoints/cli/benchmark/__init__.py +0 -0
  190. vllm/entrypoints/cli/benchmark/base.py +25 -0
  191. vllm/entrypoints/cli/benchmark/latency.py +21 -0
  192. vllm/entrypoints/cli/benchmark/main.py +58 -0
  193. vllm/entrypoints/cli/benchmark/serve.py +21 -0
  194. vllm/entrypoints/cli/benchmark/throughput.py +21 -0
  195. vllm/entrypoints/cli/collect_env.py +36 -0
  196. vllm/entrypoints/cli/main.py +60 -0
  197. vllm/entrypoints/cli/openai.py +214 -0
  198. vllm/entrypoints/cli/run_batch.py +69 -0
  199. vllm/entrypoints/cli/serve.py +232 -0
  200. vllm/entrypoints/cli/types.py +29 -0
  201. vllm/entrypoints/constants.py +10 -0
  202. vllm/entrypoints/context.py +444 -0
  203. vllm/entrypoints/harmony_utils.py +431 -0
  204. vllm/entrypoints/launcher.py +168 -0
  205. vllm/entrypoints/llm.py +1579 -0
  206. vllm/entrypoints/logger.py +79 -0
  207. vllm/entrypoints/openai/__init__.py +0 -0
  208. vllm/entrypoints/openai/api_server.py +2011 -0
  209. vllm/entrypoints/openai/cli_args.py +281 -0
  210. vllm/entrypoints/openai/logits_processors.py +90 -0
  211. vllm/entrypoints/openai/protocol.py +2590 -0
  212. vllm/entrypoints/openai/run_batch.py +497 -0
  213. vllm/entrypoints/openai/serving_chat.py +1591 -0
  214. vllm/entrypoints/openai/serving_classification.py +176 -0
  215. vllm/entrypoints/openai/serving_completion.py +688 -0
  216. vllm/entrypoints/openai/serving_embedding.py +632 -0
  217. vllm/entrypoints/openai/serving_engine.py +996 -0
  218. vllm/entrypoints/openai/serving_models.py +288 -0
  219. vllm/entrypoints/openai/serving_pooling.py +277 -0
  220. vllm/entrypoints/openai/serving_responses.py +1690 -0
  221. vllm/entrypoints/openai/serving_score.py +479 -0
  222. vllm/entrypoints/openai/serving_tokenization.py +196 -0
  223. vllm/entrypoints/openai/serving_transcription.py +136 -0
  224. vllm/entrypoints/openai/speech_to_text.py +388 -0
  225. vllm/entrypoints/openai/tool_parsers/__init__.py +51 -0
  226. vllm/entrypoints/openai/tool_parsers/abstract_tool_parser.py +164 -0
  227. vllm/entrypoints/openai/tool_parsers/deepseekv31_tool_parser.py +367 -0
  228. vllm/entrypoints/openai/tool_parsers/deepseekv3_tool_parser.py +370 -0
  229. vllm/entrypoints/openai/tool_parsers/glm4_moe_tool_parser.py +185 -0
  230. vllm/entrypoints/openai/tool_parsers/granite_20b_fc_tool_parser.py +259 -0
  231. vllm/entrypoints/openai/tool_parsers/granite_tool_parser.py +237 -0
  232. vllm/entrypoints/openai/tool_parsers/hermes_tool_parser.py +418 -0
  233. vllm/entrypoints/openai/tool_parsers/hunyuan_a13b_tool_parser.py +372 -0
  234. vllm/entrypoints/openai/tool_parsers/internlm2_tool_parser.py +216 -0
  235. vllm/entrypoints/openai/tool_parsers/jamba_tool_parser.py +308 -0
  236. vllm/entrypoints/openai/tool_parsers/kimi_k2_tool_parser.py +377 -0
  237. vllm/entrypoints/openai/tool_parsers/llama4_pythonic_tool_parser.py +316 -0
  238. vllm/entrypoints/openai/tool_parsers/llama_tool_parser.py +269 -0
  239. vllm/entrypoints/openai/tool_parsers/minimax_tool_parser.py +816 -0
  240. vllm/entrypoints/openai/tool_parsers/mistral_tool_parser.py +369 -0
  241. vllm/entrypoints/openai/tool_parsers/openai_tool_parser.py +73 -0
  242. vllm/entrypoints/openai/tool_parsers/phi4mini_tool_parser.py +112 -0
  243. vllm/entrypoints/openai/tool_parsers/pythonic_tool_parser.py +308 -0
  244. vllm/entrypoints/openai/tool_parsers/qwen3coder_tool_parser.py +707 -0
  245. vllm/entrypoints/openai/tool_parsers/seed_oss_tool_parser.py +679 -0
  246. vllm/entrypoints/openai/tool_parsers/step3_tool_parser.py +296 -0
  247. vllm/entrypoints/openai/tool_parsers/utils.py +124 -0
  248. vllm/entrypoints/openai/tool_parsers/xlam_tool_parser.py +524 -0
  249. vllm/entrypoints/renderer.py +395 -0
  250. vllm/entrypoints/score_utils.py +232 -0
  251. vllm/entrypoints/ssl.py +75 -0
  252. vllm/entrypoints/tool.py +139 -0
  253. vllm/entrypoints/tool_server.py +195 -0
  254. vllm/entrypoints/utils.py +328 -0
  255. vllm/env_override.py +23 -0
  256. vllm/envs.py +1354 -0
  257. vllm/executor/__init__.py +0 -0
  258. vllm/executor/executor_base.py +378 -0
  259. vllm/executor/mp_distributed_executor.py +244 -0
  260. vllm/executor/msgspec_utils.py +35 -0
  261. vllm/executor/multiproc_worker_utils.py +279 -0
  262. vllm/executor/ray_distributed_executor.py +699 -0
  263. vllm/executor/ray_utils.py +410 -0
  264. vllm/executor/uniproc_executor.py +152 -0
  265. vllm/forward_context.py +273 -0
  266. vllm/inputs/__init__.py +44 -0
  267. vllm/inputs/data.py +356 -0
  268. vllm/inputs/parse.py +151 -0
  269. vllm/inputs/preprocess.py +973 -0
  270. vllm/inputs/registry.py +251 -0
  271. vllm/logger.py +229 -0
  272. vllm/logging_utils/__init__.py +8 -0
  273. vllm/logging_utils/dump_input.py +81 -0
  274. vllm/logging_utils/formatter.py +79 -0
  275. vllm/logits_process.py +119 -0
  276. vllm/logprobs.py +28 -0
  277. vllm/lora/__init__.py +0 -0
  278. vllm/lora/layers/__init__.py +34 -0
  279. vllm/lora/layers/base.py +69 -0
  280. vllm/lora/layers/base_linear.py +184 -0
  281. vllm/lora/layers/column_parallel_linear.py +622 -0
  282. vllm/lora/layers/logits_processor.py +247 -0
  283. vllm/lora/layers/qkv_x_parallel_linear.py +8 -0
  284. vllm/lora/layers/replicated_linear.py +61 -0
  285. vllm/lora/layers/row_parallel_linear.py +201 -0
  286. vllm/lora/layers/utils.py +60 -0
  287. vllm/lora/layers/vocal_parallel_embedding.py +172 -0
  288. vllm/lora/lora.py +199 -0
  289. vllm/lora/models.py +792 -0
  290. vllm/lora/ops/__init__.py +0 -0
  291. vllm/lora/ops/ipex_ops/__init__.py +7 -0
  292. vllm/lora/ops/ipex_ops/lora_ops.py +44 -0
  293. vllm/lora/ops/torch_ops/__init__.py +16 -0
  294. vllm/lora/ops/torch_ops/lora_ops.py +119 -0
  295. vllm/lora/ops/triton_ops/__init__.py +12 -0
  296. vllm/lora/ops/triton_ops/kernel_utils.py +243 -0
  297. vllm/lora/ops/triton_ops/lora_expand_op.py +291 -0
  298. vllm/lora/ops/triton_ops/lora_kernel_metadata.py +148 -0
  299. vllm/lora/ops/triton_ops/lora_shrink_op.py +245 -0
  300. vllm/lora/ops/triton_ops/utils.py +126 -0
  301. vllm/lora/ops/xla_ops/__init__.py +7 -0
  302. vllm/lora/ops/xla_ops/lora_ops.py +145 -0
  303. vllm/lora/peft_helper.py +127 -0
  304. vllm/lora/punica_wrapper/__init__.py +10 -0
  305. vllm/lora/punica_wrapper/punica_base.py +458 -0
  306. vllm/lora/punica_wrapper/punica_cpu.py +349 -0
  307. vllm/lora/punica_wrapper/punica_gpu.py +279 -0
  308. vllm/lora/punica_wrapper/punica_selector.py +20 -0
  309. vllm/lora/punica_wrapper/punica_tpu.py +391 -0
  310. vllm/lora/punica_wrapper/punica_xpu.py +276 -0
  311. vllm/lora/punica_wrapper/utils.py +136 -0
  312. vllm/lora/request.py +99 -0
  313. vllm/lora/resolver.py +85 -0
  314. vllm/lora/utils.py +246 -0
  315. vllm/lora/worker_manager.py +256 -0
  316. vllm/model_executor/__init__.py +16 -0
  317. vllm/model_executor/custom_op.py +194 -0
  318. vllm/model_executor/layers/__init__.py +0 -0
  319. vllm/model_executor/layers/activation.py +575 -0
  320. vllm/model_executor/layers/attention_layer_base.py +23 -0
  321. vllm/model_executor/layers/fla/__init__.py +8 -0
  322. vllm/model_executor/layers/fla/ops/__init__.py +17 -0
  323. vllm/model_executor/layers/fla/ops/chunk.py +225 -0
  324. vllm/model_executor/layers/fla/ops/chunk_delta_h.py +290 -0
  325. vllm/model_executor/layers/fla/ops/chunk_o.py +177 -0
  326. vllm/model_executor/layers/fla/ops/chunk_scaled_dot_kkt.py +140 -0
  327. vllm/model_executor/layers/fla/ops/cumsum.py +226 -0
  328. vllm/model_executor/layers/fla/ops/fused_recurrent.py +366 -0
  329. vllm/model_executor/layers/fla/ops/index.py +39 -0
  330. vllm/model_executor/layers/fla/ops/l2norm.py +143 -0
  331. vllm/model_executor/layers/fla/ops/layernorm_guard.py +337 -0
  332. vllm/model_executor/layers/fla/ops/op.py +39 -0
  333. vllm/model_executor/layers/fla/ops/solve_tril.py +365 -0
  334. vllm/model_executor/layers/fla/ops/utils.py +180 -0
  335. vllm/model_executor/layers/fla/ops/wy_fast.py +114 -0
  336. vllm/model_executor/layers/fused_moe/__init__.py +80 -0
  337. vllm/model_executor/layers/fused_moe/batched_deep_gemm_moe.py +304 -0
  338. vllm/model_executor/layers/fused_moe/batched_triton_or_deep_gemm_moe.py +164 -0
  339. vllm/model_executor/layers/fused_moe/config.py +497 -0
  340. vllm/model_executor/layers/fused_moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  341. vllm/model_executor/layers/fused_moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  342. vllm/model_executor/layers/fused_moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  343. vllm/model_executor/layers/fused_moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  344. vllm/model_executor/layers/fused_moe/configs/E=1,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  345. vllm/model_executor/layers/fused_moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +218 -0
  346. vllm/model_executor/layers/fused_moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3.json +218 -0
  347. vllm/model_executor/layers/fused_moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  348. vllm/model_executor/layers/fused_moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  349. vllm/model_executor/layers/fused_moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  350. vllm/model_executor/layers/fused_moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  351. vllm/model_executor/layers/fused_moe/configs/E=128,N=1024,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  352. vllm/model_executor/layers/fused_moe/configs/E=128,N=1024,device_name=AMD_Instinct_MI300X.json +200 -0
  353. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  354. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  355. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H20-3e.json +146 -0
  356. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H20.json +146 -0
  357. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H200.json +146 -0
  358. vllm/model_executor/layers/fused_moe/configs/E=128,N=352,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +122 -0
  359. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  360. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  361. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  362. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  363. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  364. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20-3e.json +146 -0
  365. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20.json +146 -0
  366. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  367. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H200.json +146 -0
  368. vllm/model_executor/layers/fused_moe/configs/E=128,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  369. vllm/model_executor/layers/fused_moe/configs/E=128,N=704,device_name=NVIDIA_B200,dtype=fp8_w8a8.json +146 -0
  370. vllm/model_executor/layers/fused_moe/configs/E=128,N=704,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +114 -0
  371. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  372. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  373. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  374. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  375. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  376. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H20.json +146 -0
  377. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  378. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H200.json +146 -0
  379. vllm/model_executor/layers/fused_moe/configs/E=128,N=96,device_name=NVIDIA_H20.json +146 -0
  380. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=AMD_Instinct_MI300X.json +200 -0
  381. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=NVIDIA_B200,dtype=fp8_w8a8.json +147 -0
  382. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=NVIDIA_B200.json +146 -0
  383. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=NVIDIA_H100.json +146 -0
  384. vllm/model_executor/layers/fused_moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-40GB.json +146 -0
  385. vllm/model_executor/layers/fused_moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  386. vllm/model_executor/layers/fused_moe/configs/E=16,N=1344,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  387. vllm/model_executor/layers/fused_moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  388. vllm/model_executor/layers/fused_moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  389. vllm/model_executor/layers/fused_moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  390. vllm/model_executor/layers/fused_moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  391. vllm/model_executor/layers/fused_moe/configs/E=16,N=2688,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  392. vllm/model_executor/layers/fused_moe/configs/E=16,N=2688,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  393. vllm/model_executor/layers/fused_moe/configs/E=16,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  394. vllm/model_executor/layers/fused_moe/configs/E=16,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +146 -0
  395. vllm/model_executor/layers/fused_moe/configs/E=16,N=3200,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +130 -0
  396. vllm/model_executor/layers/fused_moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  397. vllm/model_executor/layers/fused_moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  398. vllm/model_executor/layers/fused_moe/configs/E=16,N=6400,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +130 -0
  399. vllm/model_executor/layers/fused_moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  400. vllm/model_executor/layers/fused_moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  401. vllm/model_executor/layers/fused_moe/configs/E=16,N=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +146 -0
  402. vllm/model_executor/layers/fused_moe/configs/E=16,N=800,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +130 -0
  403. vllm/model_executor/layers/fused_moe/configs/E=160,N=192,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  404. vllm/model_executor/layers/fused_moe/configs/E=160,N=192,device_name=NVIDIA_H20-3e.json +146 -0
  405. vllm/model_executor/layers/fused_moe/configs/E=160,N=320,device_name=NVIDIA_H20-3e.json +146 -0
  406. vllm/model_executor/layers/fused_moe/configs/E=160,N=640,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  407. vllm/model_executor/layers/fused_moe/configs/E=160,N=640,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  408. vllm/model_executor/layers/fused_moe/configs/E=160,N=640,device_name=NVIDIA_H100,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  409. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  410. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  411. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_H100,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  412. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  413. vllm/model_executor/layers/fused_moe/configs/E=256,N=1024,device_name=AMD_Instinct_MI325X,block_shape=[128,128].json +200 -0
  414. vllm/model_executor/layers/fused_moe/configs/E=256,N=1024,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  415. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  416. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8.json +146 -0
  417. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  418. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8.json +146 -0
  419. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  420. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  421. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  422. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  423. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  424. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  425. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  426. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  427. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  428. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  429. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  430. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  431. vllm/model_executor/layers/fused_moe/configs/E=256,N=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  432. vllm/model_executor/layers/fused_moe/configs/E=256,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  433. vllm/model_executor/layers/fused_moe/configs/E=256,N=64,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  434. vllm/model_executor/layers/fused_moe/configs/E=384,N=128,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  435. vllm/model_executor/layers/fused_moe/configs/E=384,N=128,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  436. vllm/model_executor/layers/fused_moe/configs/E=384,N=128,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  437. vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  438. vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  439. vllm/model_executor/layers/fused_moe/configs/E=40,N=2560,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  440. vllm/model_executor/layers/fused_moe/configs/E=40,N=2560,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  441. vllm/model_executor/layers/fused_moe/configs/E=40,N=2560,device_name=NVIDIA_H100,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  442. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_B200.json +146 -0
  443. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  444. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_H20-3e.json +146 -0
  445. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_H200.json +146 -0
  446. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_B200.json +146 -0
  447. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  448. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_H20-3e.json +146 -0
  449. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_H200.json +146 -0
  450. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_B200.json +146 -0
  451. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  452. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_H20-3e.json +146 -0
  453. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_H200.json +146 -0
  454. vllm/model_executor/layers/fused_moe/configs/E=512,N=64,device_name=NVIDIA_B200.json +146 -0
  455. vllm/model_executor/layers/fused_moe/configs/E=512,N=64,device_name=NVIDIA_H20-3e.json +146 -0
  456. vllm/model_executor/layers/fused_moe/configs/E=512,N=64,device_name=NVIDIA_H200.json +146 -0
  457. vllm/model_executor/layers/fused_moe/configs/E=60,N=1408,device_name=AMD_Instinct_MI300X.json +200 -0
  458. vllm/model_executor/layers/fused_moe/configs/E=60,N=176,device_name=AMD_Instinct_MI300X.json +200 -0
  459. vllm/model_executor/layers/fused_moe/configs/E=60,N=352,device_name=AMD_Instinct_MI300X.json +200 -0
  460. vllm/model_executor/layers/fused_moe/configs/E=60,N=704,device_name=AMD_Instinct_MI300X.json +200 -0
  461. vllm/model_executor/layers/fused_moe/configs/E=62,N=256,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  462. vllm/model_executor/layers/fused_moe/configs/E=62,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  463. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  464. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  465. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  466. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  467. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  468. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H200.json +146 -0
  469. vllm/model_executor/layers/fused_moe/configs/E=64,N=1536,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  470. vllm/model_executor/layers/fused_moe/configs/E=64,N=2560,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  471. vllm/model_executor/layers/fused_moe/configs/E=64,N=2560,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  472. vllm/model_executor/layers/fused_moe/configs/E=64,N=2560,device_name=NVIDIA_H200.json +146 -0
  473. vllm/model_executor/layers/fused_moe/configs/E=64,N=3072,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  474. vllm/model_executor/layers/fused_moe/configs/E=64,N=3072,device_name=NVIDIA_H20.json +146 -0
  475. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  476. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  477. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  478. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H200.json +146 -0
  479. vllm/model_executor/layers/fused_moe/configs/E=64,N=384,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  480. vllm/model_executor/layers/fused_moe/configs/E=64,N=384,device_name=NVIDIA_H20.json +146 -0
  481. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  482. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  483. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_GeForce_RTX_4090,dtype=fp8_w8a8.json +146 -0
  484. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  485. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  486. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  487. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H200.json +146 -0
  488. vllm/model_executor/layers/fused_moe/configs/E=64,N=768,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  489. vllm/model_executor/layers/fused_moe/configs/E=64,N=768,device_name=NVIDIA_H20.json +146 -0
  490. vllm/model_executor/layers/fused_moe/configs/E=64,N=896,device_name=NVIDIA_H20.json +146 -0
  491. vllm/model_executor/layers/fused_moe/configs/E=72,N=384,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  492. vllm/model_executor/layers/fused_moe/configs/E=72,N=768,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  493. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  494. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI300X.json +200 -0
  495. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  496. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI325X.json +200 -0
  497. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +138 -0
  498. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  499. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=NVIDIA_H200.json +146 -0
  500. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  501. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI300X.json +200 -0
  502. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  503. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI325X.json +200 -0
  504. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  505. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI300X.json +200 -0
  506. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  507. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI325X.json +200 -0
  508. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-40GB.json +146 -0
  509. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  510. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  511. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  512. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_H200.json +146 -0
  513. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  514. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI300X.json +200 -0
  515. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  516. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI325X.json +200 -0
  517. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  518. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  519. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  520. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +154 -0
  521. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  522. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H200.json +146 -0
  523. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  524. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI300X.json +200 -0
  525. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  526. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI325X.json +200 -0
  527. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_A100-SXM4-40GB.json +146 -0
  528. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  529. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_GeForce_RTX_4090,dtype=fp8_w8a8.json +146 -0
  530. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  531. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  532. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  533. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H200.json +146 -0
  534. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_L40S.json +173 -0
  535. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  536. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI300X.json +200 -0
  537. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  538. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI325X.json +200 -0
  539. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  540. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  541. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  542. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  543. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H200.json +146 -0
  544. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  545. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI300X.json +200 -0
  546. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  547. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI325X.json +200 -0
  548. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  549. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  550. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  551. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  552. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H200.json +146 -0
  553. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  554. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI300X.json +200 -0
  555. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  556. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI325X.json +200 -0
  557. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  558. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  559. vllm/model_executor/layers/fused_moe/configs/README +12 -0
  560. vllm/model_executor/layers/fused_moe/cpu_fused_moe.py +297 -0
  561. vllm/model_executor/layers/fused_moe/cutlass_moe.py +996 -0
  562. vllm/model_executor/layers/fused_moe/deep_gemm_moe.py +370 -0
  563. vllm/model_executor/layers/fused_moe/deep_gemm_utils.py +413 -0
  564. vllm/model_executor/layers/fused_moe/deepep_ht_prepare_finalize.py +280 -0
  565. vllm/model_executor/layers/fused_moe/deepep_ll_prepare_finalize.py +229 -0
  566. vllm/model_executor/layers/fused_moe/flashinfer_cutlass_moe.py +243 -0
  567. vllm/model_executor/layers/fused_moe/flashinfer_cutlass_prepare_finalize.py +97 -0
  568. vllm/model_executor/layers/fused_moe/fused_batched_moe.py +1042 -0
  569. vllm/model_executor/layers/fused_moe/fused_marlin_moe.py +240 -0
  570. vllm/model_executor/layers/fused_moe/fused_moe.py +2081 -0
  571. vllm/model_executor/layers/fused_moe/gpt_oss_triton_kernels_moe.py +247 -0
  572. vllm/model_executor/layers/fused_moe/layer.py +1951 -0
  573. vllm/model_executor/layers/fused_moe/modular_kernel.py +892 -0
  574. vllm/model_executor/layers/fused_moe/moe_align_block_size.py +87 -0
  575. vllm/model_executor/layers/fused_moe/moe_pallas.py +80 -0
  576. vllm/model_executor/layers/fused_moe/moe_permute_unpermute.py +205 -0
  577. vllm/model_executor/layers/fused_moe/moe_torch_iterative.py +60 -0
  578. vllm/model_executor/layers/fused_moe/pplx_prepare_finalize.py +321 -0
  579. vllm/model_executor/layers/fused_moe/prepare_finalize.py +72 -0
  580. vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py +431 -0
  581. vllm/model_executor/layers/fused_moe/routing_simulator.py +291 -0
  582. vllm/model_executor/layers/fused_moe/topk_weight_and_reduce.py +146 -0
  583. vllm/model_executor/layers/fused_moe/triton_deep_gemm_moe.py +171 -0
  584. vllm/model_executor/layers/fused_moe/trtllm_moe.py +197 -0
  585. vllm/model_executor/layers/fused_moe/utils.py +270 -0
  586. vllm/model_executor/layers/layernorm.py +381 -0
  587. vllm/model_executor/layers/lightning_attn.py +661 -0
  588. vllm/model_executor/layers/linear.py +1567 -0
  589. vllm/model_executor/layers/logits_processor.py +199 -0
  590. vllm/model_executor/layers/mamba/__init__.py +0 -0
  591. vllm/model_executor/layers/mamba/abstract.py +45 -0
  592. vllm/model_executor/layers/mamba/linear_attn.py +432 -0
  593. vllm/model_executor/layers/mamba/mamba2_metadata.py +186 -0
  594. vllm/model_executor/layers/mamba/mamba_mixer.py +517 -0
  595. vllm/model_executor/layers/mamba/mamba_mixer2.py +803 -0
  596. vllm/model_executor/layers/mamba/mamba_utils.py +202 -0
  597. vllm/model_executor/layers/mamba/ops/__init__.py +0 -0
  598. vllm/model_executor/layers/mamba/ops/causal_conv1d.py +982 -0
  599. vllm/model_executor/layers/mamba/ops/layernorm_gated.py +168 -0
  600. vllm/model_executor/layers/mamba/ops/mamba_ssm.py +414 -0
  601. vllm/model_executor/layers/mamba/ops/ssd_bmm.py +262 -0
  602. vllm/model_executor/layers/mamba/ops/ssd_chunk_scan.py +574 -0
  603. vllm/model_executor/layers/mamba/ops/ssd_chunk_state.py +751 -0
  604. vllm/model_executor/layers/mamba/ops/ssd_combined.py +248 -0
  605. vllm/model_executor/layers/mamba/ops/ssd_state_passing.py +248 -0
  606. vllm/model_executor/layers/mamba/short_conv.py +270 -0
  607. vllm/model_executor/layers/mla.py +158 -0
  608. vllm/model_executor/layers/pooler.py +732 -0
  609. vllm/model_executor/layers/quantization/__init__.py +157 -0
  610. vllm/model_executor/layers/quantization/auto_round.py +388 -0
  611. vllm/model_executor/layers/quantization/awq.py +228 -0
  612. vllm/model_executor/layers/quantization/awq_marlin.py +548 -0
  613. vllm/model_executor/layers/quantization/awq_triton.py +320 -0
  614. vllm/model_executor/layers/quantization/base_config.py +164 -0
  615. vllm/model_executor/layers/quantization/bitblas.py +464 -0
  616. vllm/model_executor/layers/quantization/bitsandbytes.py +621 -0
  617. vllm/model_executor/layers/quantization/compressed_tensors/__init__.py +0 -0
  618. vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py +795 -0
  619. vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors_moe.py +1651 -0
  620. vllm/model_executor/layers/quantization/compressed_tensors/schemes/__init__.py +27 -0
  621. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_24.py +366 -0
  622. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_scheme.py +55 -0
  623. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a16_24.py +160 -0
  624. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a16_nvfp4.py +105 -0
  625. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a4_nvfp4.py +161 -0
  626. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a8_fp8.py +169 -0
  627. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a8_int.py +135 -0
  628. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w8a16_fp8.py +121 -0
  629. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w8a8_fp8.py +156 -0
  630. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w8a8_int8.py +111 -0
  631. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_wNa16.py +201 -0
  632. vllm/model_executor/layers/quantization/compressed_tensors/transform/linear.py +227 -0
  633. vllm/model_executor/layers/quantization/compressed_tensors/transform/module.py +135 -0
  634. vllm/model_executor/layers/quantization/compressed_tensors/transform/schemes/linear_qutlass_nvfp4.py +21 -0
  635. vllm/model_executor/layers/quantization/compressed_tensors/transform/utils.py +13 -0
  636. vllm/model_executor/layers/quantization/compressed_tensors/triton_scaled_mm.py +206 -0
  637. vllm/model_executor/layers/quantization/compressed_tensors/utils.py +216 -0
  638. vllm/model_executor/layers/quantization/deepgemm.py +81 -0
  639. vllm/model_executor/layers/quantization/deepspeedfp.py +196 -0
  640. vllm/model_executor/layers/quantization/experts_int8.py +215 -0
  641. vllm/model_executor/layers/quantization/fbgemm_fp8.py +172 -0
  642. vllm/model_executor/layers/quantization/fp8.py +1179 -0
  643. vllm/model_executor/layers/quantization/gguf.py +597 -0
  644. vllm/model_executor/layers/quantization/gptq.py +300 -0
  645. vllm/model_executor/layers/quantization/gptq_bitblas.py +448 -0
  646. vllm/model_executor/layers/quantization/gptq_marlin.py +700 -0
  647. vllm/model_executor/layers/quantization/gptq_marlin_24.py +297 -0
  648. vllm/model_executor/layers/quantization/hqq_marlin.py +333 -0
  649. vllm/model_executor/layers/quantization/inc.py +61 -0
  650. vllm/model_executor/layers/quantization/input_quant_fp8.py +103 -0
  651. vllm/model_executor/layers/quantization/ipex_quant.py +410 -0
  652. vllm/model_executor/layers/quantization/kernels/__init__.py +0 -0
  653. vllm/model_executor/layers/quantization/kernels/mixed_precision/MPLinearKernel.py +91 -0
  654. vllm/model_executor/layers/quantization/kernels/mixed_precision/__init__.py +93 -0
  655. vllm/model_executor/layers/quantization/kernels/mixed_precision/allspark.py +116 -0
  656. vllm/model_executor/layers/quantization/kernels/mixed_precision/bitblas.py +302 -0
  657. vllm/model_executor/layers/quantization/kernels/mixed_precision/conch.py +92 -0
  658. vllm/model_executor/layers/quantization/kernels/mixed_precision/cutlass.py +117 -0
  659. vllm/model_executor/layers/quantization/kernels/mixed_precision/dynamic_4bit.py +92 -0
  660. vllm/model_executor/layers/quantization/kernels/mixed_precision/exllama.py +143 -0
  661. vllm/model_executor/layers/quantization/kernels/mixed_precision/machete.py +144 -0
  662. vllm/model_executor/layers/quantization/kernels/mixed_precision/marlin.py +139 -0
  663. vllm/model_executor/layers/quantization/kernels/scaled_mm/ScaledMMLinearKernel.py +67 -0
  664. vllm/model_executor/layers/quantization/kernels/scaled_mm/__init__.py +89 -0
  665. vllm/model_executor/layers/quantization/kernels/scaled_mm/aiter.py +163 -0
  666. vllm/model_executor/layers/quantization/kernels/scaled_mm/cpu.py +206 -0
  667. vllm/model_executor/layers/quantization/kernels/scaled_mm/cutlass.py +137 -0
  668. vllm/model_executor/layers/quantization/kernels/scaled_mm/triton.py +41 -0
  669. vllm/model_executor/layers/quantization/kernels/scaled_mm/xla.py +104 -0
  670. vllm/model_executor/layers/quantization/kv_cache.py +139 -0
  671. vllm/model_executor/layers/quantization/modelopt.py +1548 -0
  672. vllm/model_executor/layers/quantization/moe_wna16.py +473 -0
  673. vllm/model_executor/layers/quantization/mxfp4.py +951 -0
  674. vllm/model_executor/layers/quantization/petit.py +306 -0
  675. vllm/model_executor/layers/quantization/ptpc_fp8.py +129 -0
  676. vllm/model_executor/layers/quantization/quark/__init__.py +0 -0
  677. vllm/model_executor/layers/quantization/quark/quark.py +431 -0
  678. vllm/model_executor/layers/quantization/quark/quark_moe.py +434 -0
  679. vllm/model_executor/layers/quantization/quark/schemes/__init__.py +9 -0
  680. vllm/model_executor/layers/quantization/quark/schemes/quark_scheme.py +55 -0
  681. vllm/model_executor/layers/quantization/quark/schemes/quark_w4a4_mxfp4.py +112 -0
  682. vllm/model_executor/layers/quantization/quark/schemes/quark_w8a8_fp8.py +163 -0
  683. vllm/model_executor/layers/quantization/quark/schemes/quark_w8a8_int8.py +122 -0
  684. vllm/model_executor/layers/quantization/quark/utils.py +105 -0
  685. vllm/model_executor/layers/quantization/rtn.py +456 -0
  686. vllm/model_executor/layers/quantization/schema.py +86 -0
  687. vllm/model_executor/layers/quantization/torchao.py +214 -0
  688. vllm/model_executor/layers/quantization/tpu_int8.py +125 -0
  689. vllm/model_executor/layers/quantization/utils/__init__.py +6 -0
  690. vllm/model_executor/layers/quantization/utils/allspark_utils.py +52 -0
  691. vllm/model_executor/layers/quantization/utils/bitblas_utils.py +210 -0
  692. vllm/model_executor/layers/quantization/utils/configs/N=12288,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  693. vllm/model_executor/layers/quantization/utils/configs/N=12288,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  694. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  695. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  696. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  697. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  698. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  699. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  700. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  701. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  702. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  703. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  704. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  705. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  706. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  707. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  708. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  709. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  710. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  711. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  712. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  713. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  714. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  715. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  716. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  717. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  718. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  719. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  720. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  721. vllm/model_executor/layers/quantization/utils/configs/N=2112,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  722. vllm/model_executor/layers/quantization/utils/configs/N=2112,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  723. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  724. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  725. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  726. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  727. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  728. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  729. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  730. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  731. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  732. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=1536,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  733. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=1536,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  734. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  735. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  736. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  737. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  738. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  739. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  740. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  741. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  742. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  743. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  744. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  745. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  746. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  747. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  748. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  749. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  750. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  751. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  752. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  753. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  754. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  755. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  756. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  757. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  758. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  759. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  760. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  761. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  762. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  763. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  764. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  765. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  766. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  767. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  768. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  769. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  770. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  771. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  772. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  773. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  774. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  775. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  776. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  777. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  778. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  779. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  780. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  781. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  782. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  783. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  784. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  785. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  786. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  787. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  788. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  789. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  790. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  791. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  792. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  793. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  794. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  795. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  796. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  797. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  798. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  799. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  800. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  801. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  802. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  803. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  804. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  805. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  806. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  807. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  808. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  809. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  810. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  811. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  812. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  813. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  814. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  815. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  816. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  817. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  818. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  819. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  820. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  821. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +18 -0
  822. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  823. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  824. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  825. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  826. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  827. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  828. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  829. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  830. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  831. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  832. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  833. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  834. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  835. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  836. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  837. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  838. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  839. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  840. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  841. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  842. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  843. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  844. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  845. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  846. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  847. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  848. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  849. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  850. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  851. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  852. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  853. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  854. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  855. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  856. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  857. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  858. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  859. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  860. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  861. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  862. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  863. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  864. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  865. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  866. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  867. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  868. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  869. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  870. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  871. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  872. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  873. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  874. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  875. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  876. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  877. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  878. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  879. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  880. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  881. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  882. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  883. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  884. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  885. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  886. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  887. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  888. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  889. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  890. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  891. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  892. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  893. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  894. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  895. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  896. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=8192,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  897. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=8192,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  898. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=8192,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  899. vllm/model_executor/layers/quantization/utils/configs/N=8192,K=1536,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  900. vllm/model_executor/layers/quantization/utils/configs/N=8192,K=1536,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  901. vllm/model_executor/layers/quantization/utils/configs/N=8192,K=1536,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  902. vllm/model_executor/layers/quantization/utils/configs/README.md +3 -0
  903. vllm/model_executor/layers/quantization/utils/flashinfer_fp4_moe.py +85 -0
  904. vllm/model_executor/layers/quantization/utils/flashinfer_utils.py +258 -0
  905. vllm/model_executor/layers/quantization/utils/fp8_utils.py +795 -0
  906. vllm/model_executor/layers/quantization/utils/gptq_utils.py +96 -0
  907. vllm/model_executor/layers/quantization/utils/int8_utils.py +492 -0
  908. vllm/model_executor/layers/quantization/utils/layer_utils.py +40 -0
  909. vllm/model_executor/layers/quantization/utils/machete_utils.py +50 -0
  910. vllm/model_executor/layers/quantization/utils/marlin_utils.py +479 -0
  911. vllm/model_executor/layers/quantization/utils/marlin_utils_fp4.py +396 -0
  912. vllm/model_executor/layers/quantization/utils/marlin_utils_fp8.py +345 -0
  913. vllm/model_executor/layers/quantization/utils/marlin_utils_test.py +165 -0
  914. vllm/model_executor/layers/quantization/utils/marlin_utils_test_24.py +464 -0
  915. vllm/model_executor/layers/quantization/utils/mxfp4_utils.py +132 -0
  916. vllm/model_executor/layers/quantization/utils/mxfp8_utils.py +20 -0
  917. vllm/model_executor/layers/quantization/utils/nvfp4_emulation_utils.py +137 -0
  918. vllm/model_executor/layers/quantization/utils/nvfp4_moe_support.py +59 -0
  919. vllm/model_executor/layers/quantization/utils/petit_utils.py +122 -0
  920. vllm/model_executor/layers/quantization/utils/quant_utils.py +627 -0
  921. vllm/model_executor/layers/quantization/utils/w8a8_utils.py +458 -0
  922. vllm/model_executor/layers/resampler.py +270 -0
  923. vllm/model_executor/layers/rotary_embedding/__init__.py +190 -0
  924. vllm/model_executor/layers/rotary_embedding/base.py +156 -0
  925. vllm/model_executor/layers/rotary_embedding/common.py +105 -0
  926. vllm/model_executor/layers/rotary_embedding/deepseek_scaling_rope.py +140 -0
  927. vllm/model_executor/layers/rotary_embedding/dual_chunk_rope.py +197 -0
  928. vllm/model_executor/layers/rotary_embedding/dynamic_ntk_alpha_rope.py +41 -0
  929. vllm/model_executor/layers/rotary_embedding/dynamic_ntk_scaling_rope.py +67 -0
  930. vllm/model_executor/layers/rotary_embedding/ernie45_vl_rope.py +80 -0
  931. vllm/model_executor/layers/rotary_embedding/linear_scaling_rope.py +115 -0
  932. vllm/model_executor/layers/rotary_embedding/llama3_rope.py +54 -0
  933. vllm/model_executor/layers/rotary_embedding/llama4_vision_rope.py +81 -0
  934. vllm/model_executor/layers/rotary_embedding/mrope.py +1140 -0
  935. vllm/model_executor/layers/rotary_embedding/ntk_scaling_rope.py +42 -0
  936. vllm/model_executor/layers/rotary_embedding/phi3_long_rope_scaled_rope.py +129 -0
  937. vllm/model_executor/layers/rotary_embedding/yarn_scaling_rope.py +68 -0
  938. vllm/model_executor/layers/sampler.py +1198 -0
  939. vllm/model_executor/layers/shared_fused_moe/__init__.py +6 -0
  940. vllm/model_executor/layers/shared_fused_moe/shared_fused_moe.py +56 -0
  941. vllm/model_executor/layers/utils.py +196 -0
  942. vllm/model_executor/layers/vocab_parallel_embedding.py +487 -0
  943. vllm/model_executor/model_loader/__init__.py +138 -0
  944. vllm/model_executor/model_loader/base_loader.py +52 -0
  945. vllm/model_executor/model_loader/bitsandbytes_loader.py +787 -0
  946. vllm/model_executor/model_loader/default_loader.py +278 -0
  947. vllm/model_executor/model_loader/dummy_loader.py +28 -0
  948. vllm/model_executor/model_loader/gguf_loader.py +155 -0
  949. vllm/model_executor/model_loader/runai_streamer_loader.py +104 -0
  950. vllm/model_executor/model_loader/sharded_state_loader.py +199 -0
  951. vllm/model_executor/model_loader/tensorizer.py +743 -0
  952. vllm/model_executor/model_loader/tensorizer_loader.py +143 -0
  953. vllm/model_executor/model_loader/tpu.py +114 -0
  954. vllm/model_executor/model_loader/utils.py +271 -0
  955. vllm/model_executor/model_loader/weight_utils.py +946 -0
  956. vllm/model_executor/models/__init__.py +30 -0
  957. vllm/model_executor/models/adapters.py +542 -0
  958. vllm/model_executor/models/aimv2.py +246 -0
  959. vllm/model_executor/models/apertus.py +582 -0
  960. vllm/model_executor/models/arcee.py +423 -0
  961. vllm/model_executor/models/arctic.py +560 -0
  962. vllm/model_executor/models/aria.py +662 -0
  963. vllm/model_executor/models/aya_vision.py +470 -0
  964. vllm/model_executor/models/baichuan.py +475 -0
  965. vllm/model_executor/models/bailing_moe.py +529 -0
  966. vllm/model_executor/models/bamba.py +582 -0
  967. vllm/model_executor/models/bart.py +1343 -0
  968. vllm/model_executor/models/bert.py +613 -0
  969. vllm/model_executor/models/bert_with_rope.py +687 -0
  970. vllm/model_executor/models/blip.py +339 -0
  971. vllm/model_executor/models/blip2.py +716 -0
  972. vllm/model_executor/models/bloom.py +374 -0
  973. vllm/model_executor/models/chameleon.py +1141 -0
  974. vllm/model_executor/models/chatglm.py +479 -0
  975. vllm/model_executor/models/clip.py +407 -0
  976. vllm/model_executor/models/cohere2_vision.py +484 -0
  977. vllm/model_executor/models/commandr.py +467 -0
  978. vllm/model_executor/models/config.py +434 -0
  979. vllm/model_executor/models/constant_size_cache.py +137 -0
  980. vllm/model_executor/models/dbrx.py +473 -0
  981. vllm/model_executor/models/deepseek.py +491 -0
  982. vllm/model_executor/models/deepseek_eagle.py +241 -0
  983. vllm/model_executor/models/deepseek_mtp.py +282 -0
  984. vllm/model_executor/models/deepseek_v2.py +1058 -0
  985. vllm/model_executor/models/deepseek_vl2.py +661 -0
  986. vllm/model_executor/models/donut.py +387 -0
  987. vllm/model_executor/models/dots1.py +547 -0
  988. vllm/model_executor/models/ernie45.py +43 -0
  989. vllm/model_executor/models/ernie45_moe.py +608 -0
  990. vllm/model_executor/models/ernie45_vl.py +1510 -0
  991. vllm/model_executor/models/ernie45_vl_moe.py +728 -0
  992. vllm/model_executor/models/ernie_mtp.py +287 -0
  993. vllm/model_executor/models/exaone.py +552 -0
  994. vllm/model_executor/models/exaone4.py +535 -0
  995. vllm/model_executor/models/fairseq2_llama.py +154 -0
  996. vllm/model_executor/models/falcon.py +511 -0
  997. vllm/model_executor/models/falcon_h1.py +739 -0
  998. vllm/model_executor/models/florence2.py +1107 -0
  999. vllm/model_executor/models/fuyu.py +401 -0
  1000. vllm/model_executor/models/gemma.py +428 -0
  1001. vllm/model_executor/models/gemma2.py +425 -0
  1002. vllm/model_executor/models/gemma3.py +542 -0
  1003. vllm/model_executor/models/gemma3_mm.py +723 -0
  1004. vllm/model_executor/models/gemma3n.py +830 -0
  1005. vllm/model_executor/models/gemma3n_mm.py +767 -0
  1006. vllm/model_executor/models/glm.py +23 -0
  1007. vllm/model_executor/models/glm4.py +305 -0
  1008. vllm/model_executor/models/glm4_1v.py +1669 -0
  1009. vllm/model_executor/models/glm4_moe.py +703 -0
  1010. vllm/model_executor/models/glm4_moe_mtp.py +306 -0
  1011. vllm/model_executor/models/glm4v.py +654 -0
  1012. vllm/model_executor/models/gpt2.py +383 -0
  1013. vllm/model_executor/models/gpt_bigcode.py +346 -0
  1014. vllm/model_executor/models/gpt_j.py +340 -0
  1015. vllm/model_executor/models/gpt_neox.py +333 -0
  1016. vllm/model_executor/models/gpt_oss.py +687 -0
  1017. vllm/model_executor/models/granite.py +498 -0
  1018. vllm/model_executor/models/granite_speech.py +799 -0
  1019. vllm/model_executor/models/granitemoe.py +541 -0
  1020. vllm/model_executor/models/granitemoehybrid.py +684 -0
  1021. vllm/model_executor/models/granitemoeshared.py +342 -0
  1022. vllm/model_executor/models/gritlm.py +262 -0
  1023. vllm/model_executor/models/grok1.py +550 -0
  1024. vllm/model_executor/models/h2ovl.py +536 -0
  1025. vllm/model_executor/models/hunyuan_v1.py +937 -0
  1026. vllm/model_executor/models/hyperclovax_vision.py +1206 -0
  1027. vllm/model_executor/models/idefics2_vision_model.py +416 -0
  1028. vllm/model_executor/models/idefics3.py +758 -0
  1029. vllm/model_executor/models/interfaces.py +854 -0
  1030. vllm/model_executor/models/interfaces_base.py +195 -0
  1031. vllm/model_executor/models/intern_vit.py +481 -0
  1032. vllm/model_executor/models/internlm2.py +453 -0
  1033. vllm/model_executor/models/internlm2_ve.py +148 -0
  1034. vllm/model_executor/models/interns1.py +832 -0
  1035. vllm/model_executor/models/interns1_vit.py +418 -0
  1036. vllm/model_executor/models/internvl.py +1423 -0
  1037. vllm/model_executor/models/jais.py +374 -0
  1038. vllm/model_executor/models/jamba.py +630 -0
  1039. vllm/model_executor/models/jina_vl.py +144 -0
  1040. vllm/model_executor/models/keye.py +1684 -0
  1041. vllm/model_executor/models/keye_vl1_5.py +601 -0
  1042. vllm/model_executor/models/kimi_vl.py +620 -0
  1043. vllm/model_executor/models/lfm2.py +558 -0
  1044. vllm/model_executor/models/llama.py +671 -0
  1045. vllm/model_executor/models/llama4.py +732 -0
  1046. vllm/model_executor/models/llama4_eagle.py +241 -0
  1047. vllm/model_executor/models/llama_eagle.py +171 -0
  1048. vllm/model_executor/models/llama_eagle3.py +292 -0
  1049. vllm/model_executor/models/llava.py +872 -0
  1050. vllm/model_executor/models/llava_next.py +572 -0
  1051. vllm/model_executor/models/llava_next_video.py +479 -0
  1052. vllm/model_executor/models/llava_onevision.py +945 -0
  1053. vllm/model_executor/models/mamba.py +310 -0
  1054. vllm/model_executor/models/mamba2.py +346 -0
  1055. vllm/model_executor/models/mamba_cache.py +83 -0
  1056. vllm/model_executor/models/medusa.py +219 -0
  1057. vllm/model_executor/models/midashenglm.py +788 -0
  1058. vllm/model_executor/models/mimo.py +191 -0
  1059. vllm/model_executor/models/mimo_mtp.py +273 -0
  1060. vllm/model_executor/models/minicpm.py +593 -0
  1061. vllm/model_executor/models/minicpm3.py +230 -0
  1062. vllm/model_executor/models/minicpm_eagle.py +391 -0
  1063. vllm/model_executor/models/minicpmo.py +804 -0
  1064. vllm/model_executor/models/minicpmv.py +1786 -0
  1065. vllm/model_executor/models/minimax_cache.py +36 -0
  1066. vllm/model_executor/models/minimax_text_01.py +1027 -0
  1067. vllm/model_executor/models/minimax_vl_01.py +431 -0
  1068. vllm/model_executor/models/mistral3.py +628 -0
  1069. vllm/model_executor/models/mixtral.py +494 -0
  1070. vllm/model_executor/models/mllama.py +1697 -0
  1071. vllm/model_executor/models/mllama4.py +1079 -0
  1072. vllm/model_executor/models/mlp_speculator.py +206 -0
  1073. vllm/model_executor/models/modernbert.py +374 -0
  1074. vllm/model_executor/models/module_mapping.py +72 -0
  1075. vllm/model_executor/models/molmo.py +1569 -0
  1076. vllm/model_executor/models/moonvit.py +663 -0
  1077. vllm/model_executor/models/motif.py +345 -0
  1078. vllm/model_executor/models/mpt.py +332 -0
  1079. vllm/model_executor/models/nano_nemotron_vl.py +1395 -0
  1080. vllm/model_executor/models/nemotron.py +509 -0
  1081. vllm/model_executor/models/nemotron_h.py +633 -0
  1082. vllm/model_executor/models/nemotron_nas.py +484 -0
  1083. vllm/model_executor/models/nemotron_vl.py +655 -0
  1084. vllm/model_executor/models/nvlm_d.py +203 -0
  1085. vllm/model_executor/models/olmo.py +406 -0
  1086. vllm/model_executor/models/olmo2.py +428 -0
  1087. vllm/model_executor/models/olmoe.py +485 -0
  1088. vllm/model_executor/models/opt.py +413 -0
  1089. vllm/model_executor/models/orion.py +350 -0
  1090. vllm/model_executor/models/ovis.py +572 -0
  1091. vllm/model_executor/models/ovis2_5.py +644 -0
  1092. vllm/model_executor/models/paligemma.py +414 -0
  1093. vllm/model_executor/models/persimmon.py +345 -0
  1094. vllm/model_executor/models/phi.py +357 -0
  1095. vllm/model_executor/models/phi3.py +19 -0
  1096. vllm/model_executor/models/phi3v.py +701 -0
  1097. vllm/model_executor/models/phi4_multimodal.py +1478 -0
  1098. vllm/model_executor/models/phi4flash.py +737 -0
  1099. vllm/model_executor/models/phi4mm.py +1281 -0
  1100. vllm/model_executor/models/phi4mm_audio.py +1254 -0
  1101. vllm/model_executor/models/phi4mm_utils.py +1875 -0
  1102. vllm/model_executor/models/phimoe.py +681 -0
  1103. vllm/model_executor/models/pixtral.py +1348 -0
  1104. vllm/model_executor/models/plamo2.py +1126 -0
  1105. vllm/model_executor/models/qwen.py +363 -0
  1106. vllm/model_executor/models/qwen2.py +526 -0
  1107. vllm/model_executor/models/qwen2_5_omni_thinker.py +985 -0
  1108. vllm/model_executor/models/qwen2_5_vl.py +1256 -0
  1109. vllm/model_executor/models/qwen2_audio.py +492 -0
  1110. vllm/model_executor/models/qwen2_moe.py +558 -0
  1111. vllm/model_executor/models/qwen2_rm.py +122 -0
  1112. vllm/model_executor/models/qwen2_vl.py +1512 -0
  1113. vllm/model_executor/models/qwen3.py +344 -0
  1114. vllm/model_executor/models/qwen3_moe.py +704 -0
  1115. vllm/model_executor/models/qwen3_next.py +1298 -0
  1116. vllm/model_executor/models/qwen3_next_mtp.py +285 -0
  1117. vllm/model_executor/models/qwen_vl.py +795 -0
  1118. vllm/model_executor/models/registry.py +891 -0
  1119. vllm/model_executor/models/roberta.py +252 -0
  1120. vllm/model_executor/models/rvl.py +103 -0
  1121. vllm/model_executor/models/seed_oss.py +488 -0
  1122. vllm/model_executor/models/siglip.py +524 -0
  1123. vllm/model_executor/models/siglip2navit.py +688 -0
  1124. vllm/model_executor/models/skyworkr1v.py +914 -0
  1125. vllm/model_executor/models/smolvlm.py +44 -0
  1126. vllm/model_executor/models/solar.py +506 -0
  1127. vllm/model_executor/models/stablelm.py +344 -0
  1128. vllm/model_executor/models/starcoder2.py +357 -0
  1129. vllm/model_executor/models/step3_text.py +521 -0
  1130. vllm/model_executor/models/step3_vl.py +1091 -0
  1131. vllm/model_executor/models/swin.py +475 -0
  1132. vllm/model_executor/models/tarsier.py +649 -0
  1133. vllm/model_executor/models/telechat2.py +151 -0
  1134. vllm/model_executor/models/teleflm.py +79 -0
  1135. vllm/model_executor/models/terratorch.py +294 -0
  1136. vllm/model_executor/models/transformers.py +883 -0
  1137. vllm/model_executor/models/ultravox.py +667 -0
  1138. vllm/model_executor/models/utils.py +770 -0
  1139. vllm/model_executor/models/vision.py +125 -0
  1140. vllm/model_executor/models/voxtral.py +789 -0
  1141. vllm/model_executor/models/whisper.py +966 -0
  1142. vllm/model_executor/models/zamba2.py +1056 -0
  1143. vllm/model_executor/parameter.py +599 -0
  1144. vllm/model_executor/sampling_metadata.py +597 -0
  1145. vllm/model_executor/utils.py +97 -0
  1146. vllm/model_executor/warmup/__init__.py +0 -0
  1147. vllm/model_executor/warmup/deep_gemm_warmup.py +223 -0
  1148. vllm/model_executor/warmup/kernel_warmup.py +83 -0
  1149. vllm/multimodal/__init__.py +35 -0
  1150. vllm/multimodal/audio.py +116 -0
  1151. vllm/multimodal/base.py +219 -0
  1152. vllm/multimodal/cache.py +507 -0
  1153. vllm/multimodal/hasher.py +110 -0
  1154. vllm/multimodal/image.py +130 -0
  1155. vllm/multimodal/inputs.py +979 -0
  1156. vllm/multimodal/parse.py +496 -0
  1157. vllm/multimodal/processing.py +1921 -0
  1158. vllm/multimodal/profiling.py +313 -0
  1159. vllm/multimodal/registry.py +375 -0
  1160. vllm/multimodal/utils.py +754 -0
  1161. vllm/multimodal/video.py +312 -0
  1162. vllm/outputs.py +517 -0
  1163. vllm/platforms/__init__.py +263 -0
  1164. vllm/platforms/cpu.py +353 -0
  1165. vllm/platforms/cuda.py +731 -0
  1166. vllm/platforms/interface.py +599 -0
  1167. vllm/platforms/rocm.py +504 -0
  1168. vllm/platforms/tpu.py +236 -0
  1169. vllm/platforms/xpu.py +243 -0
  1170. vllm/plugins/__init__.py +72 -0
  1171. vllm/plugins/io_processors/__init__.py +68 -0
  1172. vllm/plugins/io_processors/interface.py +67 -0
  1173. vllm/plugins/lora_resolvers/README.md +16 -0
  1174. vllm/plugins/lora_resolvers/__init__.py +0 -0
  1175. vllm/plugins/lora_resolvers/filesystem_resolver.py +50 -0
  1176. vllm/pooling_params.py +183 -0
  1177. vllm/profiler/__init__.py +0 -0
  1178. vllm/profiler/layerwise_profile.py +375 -0
  1179. vllm/profiler/utils.py +148 -0
  1180. vllm/py.typed +2 -0
  1181. vllm/ray/__init__.py +0 -0
  1182. vllm/ray/lazy_utils.py +22 -0
  1183. vllm/ray/ray_env.py +72 -0
  1184. vllm/reasoning/__init__.py +25 -0
  1185. vllm/reasoning/abs_reasoning_parsers.py +202 -0
  1186. vllm/reasoning/deepseek_r1_reasoning_parser.py +173 -0
  1187. vllm/reasoning/glm4_moe_reasoning_parser.py +151 -0
  1188. vllm/reasoning/gptoss_reasoning_parser.py +87 -0
  1189. vllm/reasoning/granite_reasoning_parser.py +363 -0
  1190. vllm/reasoning/hunyuan_a13b_reasoning_parser.py +245 -0
  1191. vllm/reasoning/mistral_reasoning_parser.py +47 -0
  1192. vllm/reasoning/qwen3_reasoning_parser.py +151 -0
  1193. vllm/reasoning/step3_reasoning_parser.py +109 -0
  1194. vllm/sampling_params.py +577 -0
  1195. vllm/scalar_type.py +349 -0
  1196. vllm/scripts.py +15 -0
  1197. vllm/sequence.py +1465 -0
  1198. vllm/tasks.py +11 -0
  1199. vllm/test_utils.py +130 -0
  1200. vllm/third_party/__init__.py +0 -0
  1201. vllm/third_party/pynvml.py +6140 -0
  1202. vllm/tracing.py +136 -0
  1203. vllm/transformers_utils/__init__.py +24 -0
  1204. vllm/transformers_utils/chat_templates/__init__.py +5 -0
  1205. vllm/transformers_utils/chat_templates/registry.py +71 -0
  1206. vllm/transformers_utils/chat_templates/template_basic.jinja +3 -0
  1207. vllm/transformers_utils/chat_templates/template_blip2.jinja +11 -0
  1208. vllm/transformers_utils/chat_templates/template_chatml.jinja +10 -0
  1209. vllm/transformers_utils/chat_templates/template_deepseek_vl2.jinja +23 -0
  1210. vllm/transformers_utils/chat_templates/template_fuyu.jinja +3 -0
  1211. vllm/transformers_utils/chat_templates/template_minicpmv45.jinja +93 -0
  1212. vllm/transformers_utils/config.py +1043 -0
  1213. vllm/transformers_utils/config_parser_base.py +20 -0
  1214. vllm/transformers_utils/configs/__init__.py +55 -0
  1215. vllm/transformers_utils/configs/arctic.py +207 -0
  1216. vllm/transformers_utils/configs/chatglm.py +72 -0
  1217. vllm/transformers_utils/configs/deepseek_vl2.py +216 -0
  1218. vllm/transformers_utils/configs/eagle.py +84 -0
  1219. vllm/transformers_utils/configs/falcon.py +90 -0
  1220. vllm/transformers_utils/configs/jais.py +238 -0
  1221. vllm/transformers_utils/configs/kimi_vl.py +37 -0
  1222. vllm/transformers_utils/configs/medusa.py +63 -0
  1223. vllm/transformers_utils/configs/midashenglm.py +101 -0
  1224. vllm/transformers_utils/configs/mistral.py +165 -0
  1225. vllm/transformers_utils/configs/mlp_speculator.py +68 -0
  1226. vllm/transformers_utils/configs/moonvit.py +33 -0
  1227. vllm/transformers_utils/configs/nemotron.py +205 -0
  1228. vllm/transformers_utils/configs/nemotron_h.py +259 -0
  1229. vllm/transformers_utils/configs/nemotron_vl.py +56 -0
  1230. vllm/transformers_utils/configs/ovis.py +176 -0
  1231. vllm/transformers_utils/configs/qwen3_next.py +275 -0
  1232. vllm/transformers_utils/configs/speculators/__init__.py +2 -0
  1233. vllm/transformers_utils/configs/speculators/algos.py +32 -0
  1234. vllm/transformers_utils/configs/speculators/base.py +91 -0
  1235. vllm/transformers_utils/configs/step3_vl.py +123 -0
  1236. vllm/transformers_utils/configs/ultravox.py +120 -0
  1237. vllm/transformers_utils/detokenizer.py +169 -0
  1238. vllm/transformers_utils/detokenizer_utils.py +199 -0
  1239. vllm/transformers_utils/dynamic_module.py +60 -0
  1240. vllm/transformers_utils/processor.py +245 -0
  1241. vllm/transformers_utils/processors/__init__.py +16 -0
  1242. vllm/transformers_utils/processors/deepseek_vl2.py +363 -0
  1243. vllm/transformers_utils/processors/ovis.py +420 -0
  1244. vllm/transformers_utils/processors/ovis2_5.py +458 -0
  1245. vllm/transformers_utils/runai_utils.py +99 -0
  1246. vllm/transformers_utils/s3_utils.py +90 -0
  1247. vllm/transformers_utils/tokenizer.py +293 -0
  1248. vllm/transformers_utils/tokenizer_base.py +149 -0
  1249. vllm/transformers_utils/tokenizer_group.py +132 -0
  1250. vllm/transformers_utils/tokenizers/__init__.py +10 -0
  1251. vllm/transformers_utils/tokenizers/mistral.py +520 -0
  1252. vllm/transformers_utils/utils.py +99 -0
  1253. vllm/triton_utils/__init__.py +16 -0
  1254. vllm/triton_utils/importing.py +95 -0
  1255. vllm/usage/__init__.py +0 -0
  1256. vllm/usage/usage_lib.py +259 -0
  1257. vllm/utils/__init__.py +3438 -0
  1258. vllm/utils/deep_gemm.py +212 -0
  1259. vllm/utils/flashinfer.py +372 -0
  1260. vllm/utils/jsontree.py +90 -0
  1261. vllm/utils/tensor_schema.py +236 -0
  1262. vllm/v1/__init__.py +0 -0
  1263. vllm/v1/attention/__init__.py +0 -0
  1264. vllm/v1/attention/backends/__init__.py +0 -0
  1265. vllm/v1/attention/backends/cpu_attn.py +922 -0
  1266. vllm/v1/attention/backends/flash_attn.py +800 -0
  1267. vllm/v1/attention/backends/flashinfer.py +1128 -0
  1268. vllm/v1/attention/backends/flex_attention.py +796 -0
  1269. vllm/v1/attention/backends/gdn_attn.py +320 -0
  1270. vllm/v1/attention/backends/linear_attn.py +68 -0
  1271. vllm/v1/attention/backends/mamba1_attn.py +81 -0
  1272. vllm/v1/attention/backends/mamba2_attn.py +224 -0
  1273. vllm/v1/attention/backends/mamba_attn.py +52 -0
  1274. vllm/v1/attention/backends/mla/__init__.py +0 -0
  1275. vllm/v1/attention/backends/mla/common.py +1608 -0
  1276. vllm/v1/attention/backends/mla/cutlass_mla.py +301 -0
  1277. vllm/v1/attention/backends/mla/flashattn_mla.py +273 -0
  1278. vllm/v1/attention/backends/mla/flashinfer_mla.py +110 -0
  1279. vllm/v1/attention/backends/mla/flashmla.py +213 -0
  1280. vllm/v1/attention/backends/mla/rocm_aiter_mla.py +255 -0
  1281. vllm/v1/attention/backends/mla/triton_mla.py +175 -0
  1282. vllm/v1/attention/backends/pallas.py +413 -0
  1283. vllm/v1/attention/backends/rocm_aiter_fa.py +548 -0
  1284. vllm/v1/attention/backends/short_conv_attn.py +82 -0
  1285. vllm/v1/attention/backends/tree_attn.py +450 -0
  1286. vllm/v1/attention/backends/triton_attn.py +430 -0
  1287. vllm/v1/attention/backends/utils.py +834 -0
  1288. vllm/v1/attention/backends/xformers.py +437 -0
  1289. vllm/v1/core/__init__.py +0 -0
  1290. vllm/v1/core/block_pool.py +330 -0
  1291. vllm/v1/core/encoder_cache_manager.py +333 -0
  1292. vllm/v1/core/kv_cache_coordinator.py +440 -0
  1293. vllm/v1/core/kv_cache_manager.py +398 -0
  1294. vllm/v1/core/kv_cache_utils.py +1169 -0
  1295. vllm/v1/core/sched/__init__.py +0 -0
  1296. vllm/v1/core/sched/async_scheduler.py +47 -0
  1297. vllm/v1/core/sched/interface.py +158 -0
  1298. vllm/v1/core/sched/output.py +162 -0
  1299. vllm/v1/core/sched/request_queue.py +224 -0
  1300. vllm/v1/core/sched/scheduler.py +1287 -0
  1301. vllm/v1/core/sched/utils.py +69 -0
  1302. vllm/v1/core/single_type_kv_cache_manager.py +670 -0
  1303. vllm/v1/cudagraph_dispatcher.py +121 -0
  1304. vllm/v1/engine/__init__.py +202 -0
  1305. vllm/v1/engine/async_llm.py +757 -0
  1306. vllm/v1/engine/coordinator.py +357 -0
  1307. vllm/v1/engine/core.py +1245 -0
  1308. vllm/v1/engine/core_client.py +1333 -0
  1309. vllm/v1/engine/detokenizer.py +300 -0
  1310. vllm/v1/engine/exceptions.py +17 -0
  1311. vllm/v1/engine/llm_engine.py +332 -0
  1312. vllm/v1/engine/logprobs.py +201 -0
  1313. vllm/v1/engine/output_processor.py +558 -0
  1314. vllm/v1/engine/parallel_sampling.py +133 -0
  1315. vllm/v1/engine/processor.py +524 -0
  1316. vllm/v1/engine/utils.py +857 -0
  1317. vllm/v1/executor/__init__.py +0 -0
  1318. vllm/v1/executor/abstract.py +126 -0
  1319. vllm/v1/executor/multiproc_executor.py +683 -0
  1320. vllm/v1/executor/ray_distributed_executor.py +109 -0
  1321. vllm/v1/kv_cache_interface.py +275 -0
  1322. vllm/v1/metrics/__init__.py +0 -0
  1323. vllm/v1/metrics/loggers.py +717 -0
  1324. vllm/v1/metrics/prometheus.py +82 -0
  1325. vllm/v1/metrics/ray_wrappers.py +133 -0
  1326. vllm/v1/metrics/reader.py +246 -0
  1327. vllm/v1/metrics/stats.py +248 -0
  1328. vllm/v1/outputs.py +147 -0
  1329. vllm/v1/pool/__init__.py +0 -0
  1330. vllm/v1/pool/metadata.py +77 -0
  1331. vllm/v1/request.py +237 -0
  1332. vllm/v1/sample/__init__.py +0 -0
  1333. vllm/v1/sample/logits_processor/__init__.py +294 -0
  1334. vllm/v1/sample/logits_processor/builtin.py +273 -0
  1335. vllm/v1/sample/logits_processor/interface.py +97 -0
  1336. vllm/v1/sample/logits_processor/state.py +161 -0
  1337. vllm/v1/sample/metadata.py +43 -0
  1338. vllm/v1/sample/ops/__init__.py +0 -0
  1339. vllm/v1/sample/ops/bad_words.py +39 -0
  1340. vllm/v1/sample/ops/logprobs.py +26 -0
  1341. vllm/v1/sample/ops/penalties.py +43 -0
  1342. vllm/v1/sample/ops/topk_topp_sampler.py +254 -0
  1343. vllm/v1/sample/rejection_sampler.py +623 -0
  1344. vllm/v1/sample/sampler.py +281 -0
  1345. vllm/v1/sample/tpu/__init__.py +0 -0
  1346. vllm/v1/sample/tpu/metadata.py +124 -0
  1347. vllm/v1/sample/tpu/sampler.py +213 -0
  1348. vllm/v1/serial_utils.py +395 -0
  1349. vllm/v1/spec_decode/__init__.py +0 -0
  1350. vllm/v1/spec_decode/eagle.py +740 -0
  1351. vllm/v1/spec_decode/medusa.py +66 -0
  1352. vllm/v1/spec_decode/metadata.py +62 -0
  1353. vllm/v1/spec_decode/metrics.py +191 -0
  1354. vllm/v1/spec_decode/ngram_proposer.py +157 -0
  1355. vllm/v1/spec_decode/utils.py +14 -0
  1356. vllm/v1/structured_output/__init__.py +297 -0
  1357. vllm/v1/structured_output/backend_guidance.py +245 -0
  1358. vllm/v1/structured_output/backend_lm_format_enforcer.py +167 -0
  1359. vllm/v1/structured_output/backend_outlines.py +320 -0
  1360. vllm/v1/structured_output/backend_types.py +134 -0
  1361. vllm/v1/structured_output/backend_xgrammar.py +323 -0
  1362. vllm/v1/structured_output/request.py +86 -0
  1363. vllm/v1/structured_output/utils.py +373 -0
  1364. vllm/v1/utils.py +382 -0
  1365. vllm/v1/worker/__init__.py +0 -0
  1366. vllm/v1/worker/block_table.py +221 -0
  1367. vllm/v1/worker/cpu_model_runner.py +163 -0
  1368. vllm/v1/worker/cpu_worker.py +183 -0
  1369. vllm/v1/worker/gpu_input_batch.py +821 -0
  1370. vllm/v1/worker/gpu_model_runner.py +3743 -0
  1371. vllm/v1/worker/gpu_worker.py +697 -0
  1372. vllm/v1/worker/kv_connector_model_runner_mixin.py +122 -0
  1373. vllm/v1/worker/lora_model_runner_mixin.py +192 -0
  1374. vllm/v1/worker/tpu_input_batch.py +585 -0
  1375. vllm/v1/worker/tpu_model_runner.py +1947 -0
  1376. vllm/v1/worker/tpu_worker.py +340 -0
  1377. vllm/v1/worker/utils.py +290 -0
  1378. vllm/v1/worker/worker_base.py +65 -0
  1379. vllm/v1/worker/xpu_model_runner.py +53 -0
  1380. vllm/v1/worker/xpu_worker.py +179 -0
  1381. vllm/version.py +41 -0
  1382. vllm/vllm_flash_attn/.gitkeep +0 -0
  1383. vllm/worker/__init__.py +0 -0
  1384. vllm/worker/cache_engine.py +145 -0
  1385. vllm/worker/enc_dec_model_runner.py +553 -0
  1386. vllm/worker/model_runner.py +2016 -0
  1387. vllm/worker/model_runner_base.py +307 -0
  1388. vllm/worker/utils.py +49 -0
  1389. vllm/worker/worker.py +670 -0
  1390. vllm/worker/worker_base.py +651 -0
  1391. vllm_cpu_avx512vnni-0.10.2.post2.dist-info/METADATA +326 -0
  1392. vllm_cpu_avx512vnni-0.10.2.post2.dist-info/RECORD +1395 -0
  1393. vllm_cpu_avx512vnni-0.10.2.post2.dist-info/WHEEL +5 -0
  1394. vllm_cpu_avx512vnni-0.10.2.post2.dist-info/entry_points.txt +5 -0
  1395. vllm_cpu_avx512vnni-0.10.2.post2.dist-info/top_level.txt +1 -0
@@ -0,0 +1,1690 @@
1
+ # SPDX-License-Identifier: Apache-2.0
2
+ # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
+
4
+ import asyncio
5
+ import json
6
+ import time
7
+ import uuid
8
+ from collections import deque
9
+ from collections.abc import AsyncGenerator, AsyncIterator, Sequence
10
+ from contextlib import AsyncExitStack
11
+ from copy import copy
12
+ from http import HTTPStatus
13
+ from typing import Callable, Final, Optional, Union
14
+
15
+ import jinja2
16
+ import openai.types.responses as openai_responses_types
17
+ from fastapi import Request
18
+ from openai import BaseModel
19
+ # yapf conflicts with isort for this block
20
+ # yapf: disable
21
+ from openai.types.responses import (ResponseCreatedEvent,
22
+ ResponseFunctionToolCall,
23
+ ResponseInProgressEvent,
24
+ ResponseOutputItem,
25
+ ResponseOutputItemDoneEvent,
26
+ ResponseOutputMessage, ResponseOutputText,
27
+ ResponseReasoningItem,
28
+ ResponseReasoningTextDeltaEvent,
29
+ ResponseReasoningTextDoneEvent,
30
+ response_text_delta_event)
31
+ from openai.types.responses.response_output_text import (Logprob,
32
+ LogprobTopLogprob)
33
+ # yapf: enable
34
+ from openai.types.responses.response_reasoning_item import (
35
+ Content as ResponseReasoningTextContent)
36
+ from openai_harmony import Message as OpenAIHarmonyMessage
37
+
38
+ from vllm import envs
39
+ from vllm.config import ModelConfig
40
+ from vllm.engine.protocol import EngineClient
41
+ from vllm.entrypoints.chat_utils import (ChatCompletionMessageParam,
42
+ ChatTemplateContentFormatOption)
43
+ from vllm.entrypoints.context import (ConversationContext, HarmonyContext,
44
+ SimpleContext, StreamingHarmonyContext)
45
+ from vllm.entrypoints.harmony_utils import (
46
+ get_developer_message, get_stop_tokens_for_assistant_actions,
47
+ get_system_message, get_user_message, has_custom_tools,
48
+ parse_output_message, parse_remaining_state, parse_response_input,
49
+ render_for_completion)
50
+ from vllm.entrypoints.logger import RequestLogger
51
+ # yapf conflicts with isort for this block
52
+ # yapf: disable
53
+ from vllm.entrypoints.openai.protocol import (DeltaMessage, ErrorResponse,
54
+ InputTokensDetails,
55
+ OutputTokensDetails,
56
+ RequestResponseMetadata,
57
+ ResponsesRequest,
58
+ ResponsesResponse, ResponseUsage)
59
+ # yapf: enable
60
+ from vllm.entrypoints.openai.serving_engine import OpenAIServing
61
+ from vllm.entrypoints.openai.serving_models import OpenAIServingModels
62
+ from vllm.entrypoints.tool_server import ToolServer
63
+ from vllm.inputs.data import TokensPrompt as EngineTokensPrompt
64
+ from vllm.logger import init_logger
65
+ from vllm.logprobs import Logprob as SampleLogprob
66
+ from vllm.logprobs import SampleLogprobs
67
+ from vllm.outputs import CompletionOutput
68
+ from vllm.reasoning import ReasoningParser, ReasoningParserManager
69
+ from vllm.sampling_params import SamplingParams
70
+ from vllm.transformers_utils.tokenizer import AnyTokenizer
71
+ from vllm.utils import random_uuid
72
+
73
+ logger = init_logger(__name__)
74
+
75
+
76
+ class OpenAIServingResponses(OpenAIServing):
77
+
78
+ def __init__(
79
+ self,
80
+ engine_client: EngineClient,
81
+ model_config: ModelConfig,
82
+ models: OpenAIServingModels,
83
+ *,
84
+ request_logger: Optional[RequestLogger],
85
+ chat_template: Optional[str],
86
+ chat_template_content_format: ChatTemplateContentFormatOption,
87
+ return_tokens_as_token_ids: bool = False,
88
+ reasoning_parser: str = "",
89
+ enable_auto_tools: bool = False,
90
+ tool_parser: Optional[str] = None,
91
+ tool_server: Optional[ToolServer] = None,
92
+ enable_prompt_tokens_details: bool = False,
93
+ enable_force_include_usage: bool = False,
94
+ enable_log_outputs: bool = False,
95
+ log_error_stack: bool = False,
96
+ ) -> None:
97
+ super().__init__(
98
+ engine_client=engine_client,
99
+ model_config=model_config,
100
+ models=models,
101
+ request_logger=request_logger,
102
+ return_tokens_as_token_ids=return_tokens_as_token_ids,
103
+ enable_force_include_usage=enable_force_include_usage,
104
+ log_error_stack=log_error_stack,
105
+ )
106
+
107
+ self.chat_template = chat_template
108
+ self.chat_template_content_format: Final = chat_template_content_format
109
+ self.enable_log_outputs = enable_log_outputs
110
+
111
+ self.reasoning_parser: Optional[Callable[[AnyTokenizer],
112
+ ReasoningParser]] = None
113
+ if reasoning_parser:
114
+ try:
115
+ self.reasoning_parser = (
116
+ ReasoningParserManager.get_reasoning_parser(
117
+ reasoning_parser))
118
+ assert self.reasoning_parser is not None
119
+ except Exception as e:
120
+ raise TypeError(
121
+ f"{reasoning_parser=} has not been registered") from e
122
+
123
+ self.enable_prompt_tokens_details = enable_prompt_tokens_details
124
+ self.enable_force_include_usage = enable_force_include_usage
125
+ self.default_sampling_params = (
126
+ self.model_config.get_diff_sampling_param())
127
+ if self.default_sampling_params:
128
+ source = self.model_config.generation_config
129
+ source = "model" if source == "auto" else source
130
+ logger.info("Using default chat sampling params from %s: %s",
131
+ source, self.default_sampling_params)
132
+
133
+ # If False (default), the "store" option is (silently) ignored and the
134
+ # response is not stored. If True, the response is stored in memory.
135
+ # NOTE(woosuk): This may not be intuitive for users, as the default
136
+ # behavior in OpenAI's Responses API is to store the response, but
137
+ # vLLM's default behavior is not.
138
+ self.enable_store = envs.VLLM_ENABLE_RESPONSES_API_STORE
139
+ if self.enable_store:
140
+ logger.warning_once(
141
+ "`VLLM_ENABLE_RESPONSES_API_STORE` is enabled. This may "
142
+ "cause a memory leak since we never remove responses from "
143
+ "the store.")
144
+
145
+ self.use_harmony = model_config.hf_config.model_type == "gpt_oss"
146
+ if self.use_harmony:
147
+ logger.warning("For gpt-oss, we ignore --enable-auto-tool-choice "
148
+ "and always enable tool use.")
149
+ # OpenAI models have two EOS-like tokens: <|return|> and <|call|>.
150
+ # We need to add them to the stop token ids.
151
+ if "stop_token_ids" not in self.default_sampling_params:
152
+ self.default_sampling_params["stop_token_ids"] = []
153
+ self.default_sampling_params["stop_token_ids"].extend(
154
+ get_stop_tokens_for_assistant_actions())
155
+
156
+ # set up tool use
157
+ self.enable_auto_tools: bool = enable_auto_tools
158
+ if self.enable_auto_tools:
159
+ logger.info(
160
+ "\"auto\" tool choice has been enabled please note that while"
161
+ " the parallel_tool_calls client option is preset for "
162
+ "compatibility reasons, it will be ignored.")
163
+
164
+ # HACK(woosuk): This is a hack. We should use a better store.
165
+ # FIXME: If enable_store=True, this may cause a memory leak since we
166
+ # never remove responses from the store.
167
+ self.response_store: dict[str, ResponsesResponse] = {}
168
+ self.response_store_lock = asyncio.Lock()
169
+
170
+ # HACK(woosuk): This is a hack. We should use a better store.
171
+ # FIXME: If enable_store=True, this may cause a memory leak since we
172
+ # never remove messages from the store.
173
+ self.msg_store: dict[str, list[ChatCompletionMessageParam]] = {}
174
+
175
+ # HACK(wuhang): This is a hack. We should use a better store.
176
+ # FIXME: If enable_store=True, this may cause a memory leak since we
177
+ # never remove events from the store.
178
+ self.event_store: dict[str, tuple[deque[str], asyncio.Event]] = {}
179
+
180
+ self.background_tasks: dict[str, asyncio.Task] = {}
181
+
182
+ self.tool_server = tool_server
183
+
184
+ async def create_responses(
185
+ self,
186
+ request: ResponsesRequest,
187
+ raw_request: Optional[Request] = None,
188
+ ) -> Union[AsyncGenerator[str, None], ResponsesResponse, ErrorResponse]:
189
+ error_check_ret = await self._check_model(request)
190
+ if error_check_ret is not None:
191
+ logger.error("Error with model %s", error_check_ret)
192
+ return error_check_ret
193
+
194
+ # If the engine is dead, raise the engine's DEAD_ERROR.
195
+ # This is required for the streaming case, where we return a
196
+ # success status before we actually start generating text :).
197
+ if self.engine_client.errored:
198
+ raise self.engine_client.dead_error
199
+
200
+ if request.store and not self.enable_store:
201
+ if request.background:
202
+ return self.create_error_response(
203
+ err_type="invalid_request_error",
204
+ message=(
205
+ "This vLLM engine does not support `store=True` and "
206
+ "therefore does not support the background mode. To "
207
+ "enable these features, set the environment variable "
208
+ "`VLLM_ENABLE_RESPONSES_API_STORE=1` when launching "
209
+ "the vLLM server."),
210
+ status_code=HTTPStatus.BAD_REQUEST,
211
+ )
212
+ # Disable the store option.
213
+ # NOTE(woosuk): Although returning an error is possible, we opted
214
+ # to implicitly disable store and process the request anyway, as
215
+ # we assume most users do not intend to actually store the response
216
+ # (i.e., their request's `store=True` just because it's the default
217
+ # value).
218
+ request.store = False
219
+ if self.use_harmony and request.is_include_output_logprobs():
220
+ return self.create_error_response(
221
+ err_type="invalid_request_error",
222
+ message="logprobs are not supported with gpt-oss models",
223
+ status_code=HTTPStatus.BAD_REQUEST,
224
+ )
225
+
226
+ # Handle the previous response ID.
227
+ prev_response_id = request.previous_response_id
228
+ if prev_response_id is not None:
229
+ if not prev_response_id.startswith("resp_"):
230
+ return self._make_invalid_id_error(prev_response_id)
231
+ async with self.response_store_lock:
232
+ prev_response = self.response_store.get(prev_response_id)
233
+ if prev_response is None:
234
+ return self._make_not_found_error(prev_response_id)
235
+ else:
236
+ prev_response = None
237
+
238
+ try:
239
+ lora_request = self._maybe_get_adapters(request)
240
+ model_name = self.models.model_name(lora_request)
241
+ tokenizer = await self.engine_client.get_tokenizer(lora_request)
242
+
243
+ if self.use_harmony:
244
+ messages, request_prompts, engine_prompts = (
245
+ self._make_request_with_harmony(request, prev_response))
246
+ else:
247
+ messages, request_prompts, engine_prompts = (
248
+ await self._make_request(request, prev_response,
249
+ tokenizer))
250
+
251
+ except (ValueError, TypeError, RuntimeError, jinja2.TemplateError,
252
+ NotImplementedError) as e:
253
+ logger.exception("Error in preprocessing prompt inputs")
254
+ return self.create_error_response(f"{e} {e.__cause__}")
255
+
256
+ request_metadata = RequestResponseMetadata(
257
+ request_id=request.request_id)
258
+ if raw_request:
259
+ raw_request.state.request_metadata = request_metadata
260
+
261
+ # Schedule the request and get the result generator.
262
+ generators: list[AsyncGenerator[ConversationContext, None]] = []
263
+
264
+ builtin_tool_list: list[str] = []
265
+ if self.use_harmony and self.tool_server is not None:
266
+ if self.tool_server.has_tool("browser"):
267
+ builtin_tool_list.append("browser")
268
+ if self.tool_server.has_tool("python"):
269
+ builtin_tool_list.append("python")
270
+ if self.tool_server.has_tool("container"):
271
+ builtin_tool_list.append("container")
272
+
273
+ if self.tool_server is not None:
274
+ available_tools = builtin_tool_list
275
+ else:
276
+ assert len(builtin_tool_list) == 0
277
+ available_tools = []
278
+ try:
279
+ for i, engine_prompt in enumerate(engine_prompts):
280
+ default_max_tokens = self.max_model_len - len(
281
+ engine_prompt["prompt_token_ids"])
282
+ sampling_params = request.to_sampling_params(
283
+ default_max_tokens, self.default_sampling_params)
284
+
285
+ trace_headers = (None if raw_request is None else await
286
+ self._get_trace_headers(raw_request.headers))
287
+
288
+ context: ConversationContext
289
+ if self.use_harmony:
290
+ if request.stream:
291
+ context = StreamingHarmonyContext(
292
+ messages, available_tools)
293
+ else:
294
+ context = HarmonyContext(messages, available_tools)
295
+ else:
296
+ context = SimpleContext()
297
+ generator = self._generate_with_builtin_tools(
298
+ request_id=request.request_id,
299
+ request_prompt=request_prompts[i],
300
+ engine_prompt=engine_prompt,
301
+ sampling_params=sampling_params,
302
+ context=context,
303
+ lora_request=lora_request,
304
+ priority=request.priority,
305
+ trace_headers=trace_headers,
306
+ )
307
+ generators.append(generator)
308
+ except ValueError as e:
309
+ # TODO: Use a vllm-specific Validation Error
310
+ return self.create_error_response(str(e))
311
+
312
+ assert len(generators) == 1
313
+ result_generator, = generators
314
+
315
+ # Store the input messages.
316
+ if request.store:
317
+ self.msg_store[request.request_id] = messages
318
+
319
+ if request.background:
320
+ created_time = int(time.time())
321
+ response = ResponsesResponse.from_request(
322
+ request,
323
+ sampling_params,
324
+ model_name=model_name,
325
+ created_time=created_time,
326
+ output=[],
327
+ status="queued",
328
+ usage=None,
329
+ )
330
+ async with self.response_store_lock:
331
+ self.response_store[response.id] = response
332
+
333
+ # Run the request in the background.
334
+ if request.stream:
335
+ task = asyncio.create_task(
336
+ self._run_background_request_stream(
337
+ request,
338
+ sampling_params,
339
+ result_generator,
340
+ context,
341
+ model_name,
342
+ tokenizer,
343
+ request_metadata,
344
+ created_time,
345
+ ),
346
+ name=f"create_{request.request_id}",
347
+ )
348
+ else:
349
+ task = asyncio.create_task(
350
+ self._run_background_request(
351
+ request,
352
+ sampling_params,
353
+ result_generator,
354
+ context,
355
+ model_name,
356
+ tokenizer,
357
+ request_metadata,
358
+ created_time,
359
+ ),
360
+ name=f"create_{response.id}",
361
+ )
362
+
363
+ # For cleanup.
364
+ response_id = response.id
365
+ self.background_tasks[response_id] = task
366
+ task.add_done_callback(
367
+ lambda _: self.background_tasks.pop(response_id, None))
368
+
369
+ if request.stream:
370
+ return self.responses_background_stream_generator(
371
+ request.request_id)
372
+ return response
373
+
374
+ if request.stream:
375
+ return self.responses_stream_generator(
376
+ request,
377
+ sampling_params,
378
+ result_generator,
379
+ context,
380
+ model_name,
381
+ tokenizer,
382
+ request_metadata,
383
+ )
384
+
385
+ try:
386
+ return await self.responses_full_generator(
387
+ request,
388
+ sampling_params,
389
+ result_generator,
390
+ context,
391
+ model_name,
392
+ tokenizer,
393
+ request_metadata,
394
+ )
395
+ except Exception as e:
396
+ return self.create_error_response(str(e))
397
+
398
+ async def _make_request(
399
+ self,
400
+ request: ResponsesRequest,
401
+ prev_response: Optional[ResponsesResponse],
402
+ tokenizer: AnyTokenizer,
403
+ ):
404
+ if len(request.tools) > 0:
405
+ raise NotImplementedError(
406
+ "Tool use is not supported in Responses API without Harmony")
407
+ # Construct the input messages.
408
+ messages = self._construct_input_messages(request, prev_response)
409
+ _, request_prompts, engine_prompts = await self._preprocess_chat(
410
+ request,
411
+ tokenizer,
412
+ messages,
413
+ chat_template=self.chat_template,
414
+ chat_template_content_format=self.chat_template_content_format,
415
+ )
416
+ return messages, request_prompts, engine_prompts
417
+
418
+ def _make_request_with_harmony(
419
+ self,
420
+ request: ResponsesRequest,
421
+ prev_response: Optional[ResponsesResponse],
422
+ ):
423
+ if request.tool_choice != "auto":
424
+ raise NotImplementedError(
425
+ "Only 'auto' tool_choice is supported in "
426
+ "response API with Harmony")
427
+ messages = self._construct_input_messages_with_harmony(
428
+ request, prev_response)
429
+ prompt_token_ids = render_for_completion(messages)
430
+ engine_prompt = EngineTokensPrompt(prompt_token_ids=prompt_token_ids)
431
+
432
+ # Add cache_salt if provided in the request
433
+ if request.cache_salt is not None:
434
+ engine_prompt["cache_salt"] = request.cache_salt
435
+
436
+ return messages, [prompt_token_ids], [engine_prompt]
437
+
438
+ async def responses_full_generator(
439
+ self,
440
+ request: ResponsesRequest,
441
+ sampling_params: SamplingParams,
442
+ result_generator: AsyncIterator[ConversationContext],
443
+ context: ConversationContext,
444
+ model_name: str,
445
+ tokenizer: AnyTokenizer,
446
+ request_metadata: RequestResponseMetadata,
447
+ created_time: Optional[int] = None,
448
+ ) -> Union[ErrorResponse, ResponsesResponse]:
449
+ if created_time is None:
450
+ created_time = int(time.time())
451
+
452
+ async with AsyncExitStack() as exit_stack:
453
+ try:
454
+ await context.init_tool_sessions(self.tool_server, exit_stack,
455
+ request.request_id)
456
+ async for _ in result_generator:
457
+ pass
458
+ except asyncio.CancelledError:
459
+ return self.create_error_response("Client disconnected")
460
+ except ValueError as e:
461
+ # TODO: Use a vllm-specific Validation Error
462
+ return self.create_error_response(str(e))
463
+
464
+ if self.use_harmony:
465
+ assert isinstance(context, HarmonyContext)
466
+ output = self._make_response_output_items_with_harmony(context)
467
+ num_tool_output_tokens = context.num_tool_output_tokens
468
+ else:
469
+ assert isinstance(context, SimpleContext)
470
+ final_res = context.last_output
471
+ assert final_res is not None
472
+ assert len(final_res.outputs) == 1
473
+ final_output = final_res.outputs[0]
474
+
475
+ output = self._make_response_output_items(request, final_output,
476
+ tokenizer)
477
+
478
+ # Calculate usage.
479
+ assert final_res.prompt_token_ids is not None
480
+ num_tool_output_tokens = 0
481
+
482
+ assert isinstance(context, (SimpleContext, HarmonyContext))
483
+ num_prompt_tokens = context.num_prompt_tokens
484
+ num_generated_tokens = context.num_output_tokens
485
+ num_cached_tokens = context.num_cached_tokens
486
+ num_reasoning_tokens = context.num_reasoning_tokens
487
+
488
+ usage = ResponseUsage(
489
+ input_tokens=num_prompt_tokens,
490
+ output_tokens=num_generated_tokens,
491
+ total_tokens=num_prompt_tokens + num_generated_tokens,
492
+ input_tokens_details=InputTokensDetails(
493
+ cached_tokens=num_cached_tokens),
494
+ output_tokens_details=OutputTokensDetails(
495
+ reasoning_tokens=num_reasoning_tokens,
496
+ tool_output_tokens=num_tool_output_tokens),
497
+ )
498
+ response = ResponsesResponse.from_request(
499
+ request,
500
+ sampling_params,
501
+ model_name=model_name,
502
+ created_time=created_time,
503
+ output=output,
504
+ status="completed",
505
+ usage=usage,
506
+ )
507
+
508
+ if request.store:
509
+ async with self.response_store_lock:
510
+ stored_response = self.response_store.get(response.id)
511
+ # If the response is already cancelled, don't update it.
512
+ if (stored_response is None
513
+ or stored_response.status != "cancelled"):
514
+ self.response_store[response.id] = response
515
+ return response
516
+
517
+ def _topk_logprobs(self, logprobs: dict[int,
518
+ SampleLogprob], top_logprobs: int,
519
+ tokenizer: AnyTokenizer) -> list[LogprobTopLogprob]:
520
+ """Returns the top-k logprobs from the logprobs dictionary."""
521
+ out = []
522
+ for i, (token_id, _logprob) in enumerate(logprobs.items()):
523
+ if i >= top_logprobs:
524
+ break
525
+ text = _logprob.decoded_token if _logprob.decoded_token \
526
+ is not None else tokenizer.decode([token_id])
527
+ out.append(
528
+ LogprobTopLogprob(
529
+ token=text,
530
+ logprob=max(_logprob.logprob, -9999.0),
531
+ bytes=list(text.encode("utf-8", errors="replace")),
532
+ ))
533
+ return out
534
+
535
+ def _create_response_logprobs(
536
+ self,
537
+ token_ids: Sequence[int],
538
+ logprobs: Optional[SampleLogprobs],
539
+ tokenizer: AnyTokenizer,
540
+ top_logprobs: Optional[int] = None) -> list[Logprob]:
541
+ assert logprobs is not None, "logprobs must be provided"
542
+ assert len(token_ids) == len(logprobs), (
543
+ "token_ids and logprobs.token_ids must have the same length")
544
+ out = []
545
+ for i, token_id in enumerate(token_ids):
546
+ logprob = logprobs[i]
547
+ token_logprob = logprob[token_id]
548
+ text = token_logprob.decoded_token if token_logprob.decoded_token \
549
+ is not None else tokenizer.decode([token_id])
550
+ out.append(
551
+ Logprob(
552
+ token=text,
553
+ logprob=max(token_logprob.logprob, -9999.0),
554
+ bytes=list(text.encode("utf-8", errors="replace")),
555
+ top_logprobs=self._topk_logprobs(logprob,
556
+ top_logprobs=top_logprobs,
557
+ tokenizer=tokenizer)
558
+ if top_logprobs else [],
559
+ ))
560
+ return out
561
+
562
+ def _create_stream_response_logprobs(
563
+ self,
564
+ token_ids: Sequence[int],
565
+ logprobs: Optional[SampleLogprobs],
566
+ tokenizer: AnyTokenizer,
567
+ top_logprobs: Optional[int] = None
568
+ ) -> list[response_text_delta_event.Logprob]:
569
+ lgs = self._create_response_logprobs(token_ids=token_ids,
570
+ logprobs=logprobs,
571
+ tokenizer=tokenizer,
572
+ top_logprobs=top_logprobs)
573
+ return [
574
+ response_text_delta_event.Logprob(
575
+ token=lg.token,
576
+ logprob=lg.logprob,
577
+ top_logprobs=[
578
+ response_text_delta_event.LogprobTopLogprob(
579
+ token=tl.token, logprob=tl.logprob)
580
+ for tl in lg.top_logprobs
581
+ ]) for lg in lgs
582
+ ]
583
+
584
+ def _make_response_output_items(
585
+ self,
586
+ request: ResponsesRequest,
587
+ final_output: CompletionOutput,
588
+ tokenizer: AnyTokenizer,
589
+ ) -> list[ResponseOutputItem]:
590
+ if self.reasoning_parser:
591
+ try:
592
+ reasoning_parser = self.reasoning_parser(tokenizer)
593
+ except RuntimeError as e:
594
+ logger.exception("Error in reasoning parser creation.")
595
+ raise e
596
+
597
+ reasoning_content, content = (
598
+ reasoning_parser.extract_reasoning_content(final_output.text,
599
+ request=request))
600
+ else:
601
+ reasoning_content = None
602
+ content = final_output.text
603
+
604
+ # Log complete response if output logging is enabled
605
+ if self.enable_log_outputs and self.request_logger:
606
+ output_text = ""
607
+ if content:
608
+ output_text = content
609
+ elif reasoning_content:
610
+ output_text = f"[reasoning: {reasoning_content}]"
611
+
612
+ if output_text:
613
+ self.request_logger.log_outputs(
614
+ request_id=request.request_id,
615
+ outputs=output_text,
616
+ output_token_ids=final_output.token_ids,
617
+ finish_reason=final_output.finish_reason,
618
+ is_streaming=False,
619
+ delta=False,
620
+ )
621
+
622
+ output = []
623
+ if reasoning_content:
624
+ reasoning_item = ResponseReasoningItem(
625
+ id=f"rs_{random_uuid()}",
626
+ summary=[],
627
+ type="reasoning",
628
+ content=[
629
+ ResponseReasoningTextContent(text=reasoning_content,
630
+ type="reasoning_text")
631
+ ],
632
+ status=None, # NOTE: Only the last output item has status.
633
+ )
634
+ output.append(reasoning_item)
635
+ if content:
636
+ output_text = ResponseOutputText(
637
+ text=content,
638
+ annotations=[], # TODO
639
+ type="output_text",
640
+ logprobs=self._create_response_logprobs(
641
+ token_ids=final_output.token_ids,
642
+ logprobs=final_output.logprobs,
643
+ tokenizer=tokenizer,
644
+ top_logprobs=request.top_logprobs,
645
+ ) if request.is_include_output_logprobs() else None,
646
+ )
647
+ message = ResponseOutputMessage(
648
+ id=f"msg_{random_uuid()}",
649
+ content=[output_text],
650
+ role="assistant",
651
+ status="completed",
652
+ type="message",
653
+ )
654
+ output.append(message)
655
+ return output
656
+
657
+ def _make_response_output_items_with_harmony(
658
+ self,
659
+ context: HarmonyContext,
660
+ ) -> list[ResponseOutputItem]:
661
+ output_items = []
662
+ num_init_messages = context.num_init_messages
663
+ for msg in context.messages[num_init_messages:]:
664
+ output_items.extend(parse_output_message(msg))
665
+ # Handle the generation stopped in the middle (if any).
666
+ last_items = parse_remaining_state(context.parser)
667
+ if last_items:
668
+ output_items.extend(last_items)
669
+ return output_items
670
+
671
+ def _construct_input_messages(
672
+ self,
673
+ request: ResponsesRequest,
674
+ prev_response: Optional[ResponsesResponse] = None,
675
+ ) -> list[ChatCompletionMessageParam]:
676
+ messages: list[ChatCompletionMessageParam] = []
677
+ if request.instructions:
678
+ messages.append({
679
+ "role": "system",
680
+ "content": request.instructions,
681
+ })
682
+
683
+ # Prepend the conversation history.
684
+ if prev_response is not None:
685
+ # Add the previous messages.
686
+ prev_msg = self.msg_store[prev_response.id]
687
+ messages.extend(prev_msg)
688
+
689
+ # Add the previous output.
690
+ for output_item in prev_response.output:
691
+ # NOTE: We skip the reasoning output.
692
+ if isinstance(output_item, ResponseOutputMessage):
693
+ for content in output_item.content:
694
+ messages.append({
695
+ "role": "assistant",
696
+ "content": content.text,
697
+ })
698
+
699
+ # Append the new input.
700
+ # Responses API supports simple text inputs without chat format.
701
+ if isinstance(request.input, str):
702
+ messages.append({"role": "user", "content": request.input})
703
+ else:
704
+ messages.extend(request.input) # type: ignore
705
+ return messages
706
+
707
+ def _construct_input_messages_with_harmony(
708
+ self,
709
+ request: ResponsesRequest,
710
+ prev_response: Optional[ResponsesResponse],
711
+ ) -> list[OpenAIHarmonyMessage]:
712
+ messages: list[OpenAIHarmonyMessage] = []
713
+ if prev_response is None:
714
+ # New conversation.
715
+ reasoning_effort = (request.reasoning.effort
716
+ if request.reasoning else None)
717
+ # Temporary: OpenAI types doesn't have container tool
718
+ # so we used MCP to cover that, up for change
719
+ tool_types = [tool.type for tool in request.tools]
720
+ if envs.VLLM_GPT_OSS_USE_CONTAINER_TOOL:
721
+ tool_types.append("container")
722
+ enable_browser = ("web_search_preview" in tool_types
723
+ and self.tool_server is not None
724
+ and self.tool_server.has_tool("browser"))
725
+ enable_code_interpreter = ("code_interpreter" in tool_types
726
+ and self.tool_server is not None
727
+ and self.tool_server.has_tool("python"))
728
+ enable_container = ("container" in tool_types
729
+ and self.tool_server is not None
730
+ and self.tool_server.has_tool("container"))
731
+ with_custom_tools = has_custom_tools(tool_types)
732
+ sys_msg = get_system_message(
733
+ reasoning_effort=reasoning_effort,
734
+ browser_description=self.tool_server.get_tool_description(
735
+ "browser")
736
+ if enable_browser and self.tool_server is not None else None,
737
+ python_description=self.tool_server.get_tool_description(
738
+ "python") if enable_code_interpreter
739
+ and self.tool_server is not None else None,
740
+ container_description=self.tool_server.get_tool_description(
741
+ "container")
742
+ if enable_container and self.tool_server is not None else None,
743
+ instructions=request.instructions,
744
+ with_custom_tools=with_custom_tools,
745
+ )
746
+ messages.append(sys_msg)
747
+ if with_custom_tools:
748
+ dev_msg = get_developer_message(
749
+ instructions=request.instructions, tools=request.tools)
750
+ messages.append(dev_msg)
751
+ else:
752
+ # Continue the previous conversation.
753
+ # FIXME(woosuk): Currently, request params like reasoning and
754
+ # instructions are ignored.
755
+ prev_msgs = self.msg_store[prev_response.id]
756
+ # Remove the previous chain-of-thoughts if there is a new "final"
757
+ # message. Note that this also removes these messages from the
758
+ # msg_store.
759
+ if len(prev_msgs) > 0:
760
+ last_msg = prev_msgs[-1]
761
+ assert isinstance(last_msg, OpenAIHarmonyMessage)
762
+ if last_msg.channel == "final":
763
+ prev_final_msg_idx = -1
764
+ for i in range(len(prev_msgs) - 2, -1, -1):
765
+ prev_msg_i = prev_msgs[i]
766
+ assert isinstance(prev_msg_i, OpenAIHarmonyMessage)
767
+ if prev_msg_i.channel == "final":
768
+ prev_final_msg_idx = i
769
+ break
770
+ recent_turn_msgs = prev_msgs[prev_final_msg_idx + 1:]
771
+ del prev_msgs[prev_final_msg_idx + 1:]
772
+ for msg in recent_turn_msgs:
773
+ assert isinstance(msg, OpenAIHarmonyMessage)
774
+ if msg.channel != "analysis":
775
+ prev_msgs.append(msg)
776
+ messages.extend(prev_msgs)
777
+ # Append the new input.
778
+ # Responses API supports simple text inputs without chat format.
779
+ if isinstance(request.input, str):
780
+ messages.append(get_user_message(request.input))
781
+ else:
782
+ if prev_response is not None:
783
+ prev_outputs = copy(prev_response.output)
784
+ else:
785
+ prev_outputs = []
786
+ for response_msg in request.input:
787
+ messages.append(
788
+ parse_response_input(response_msg, prev_outputs))
789
+ # User passes in a tool call request and its output. We need
790
+ # to add the tool call request to prev_outputs so that the
791
+ # parse_response_input can find the tool call request when
792
+ # parsing the tool call output.
793
+ if isinstance(response_msg, ResponseFunctionToolCall):
794
+ prev_outputs.append(response_msg)
795
+ return messages
796
+
797
+ async def _run_background_request_stream(
798
+ self,
799
+ request: ResponsesRequest,
800
+ *args,
801
+ **kwargs,
802
+ ):
803
+ event_deque: deque[str] = deque()
804
+ new_event_signal = asyncio.Event()
805
+ self.event_store[request.request_id] = (event_deque, new_event_signal)
806
+ response = None
807
+ try:
808
+ generator = self.responses_stream_generator(
809
+ request, *args, **kwargs)
810
+ async for event in generator:
811
+ event_deque.append(event)
812
+ new_event_signal.set() # Signal new event available
813
+ except Exception as e:
814
+ logger.exception("Background request failed for %s",
815
+ request.request_id)
816
+ response = self.create_error_response(str(e))
817
+ finally:
818
+ # Mark as finished with a special marker
819
+ event_deque.append("__STREAM_END__")
820
+ new_event_signal.set()
821
+
822
+ if response is not None and isinstance(response, ErrorResponse):
823
+ # If the request has failed, update the status to "failed".
824
+ response_id = request.request_id
825
+ async with self.response_store_lock:
826
+ stored_response = self.response_store.get(response_id)
827
+ assert stored_response is not None
828
+ if stored_response.status not in ("completed", "cancelled"):
829
+ stored_response.status = "failed"
830
+
831
+ async def _run_background_request(
832
+ self,
833
+ request: ResponsesRequest,
834
+ *args,
835
+ **kwargs,
836
+ ):
837
+ try:
838
+ response = await self.responses_full_generator(
839
+ request, *args, **kwargs)
840
+ except Exception as e:
841
+ logger.exception("Background request failed for %s",
842
+ request.request_id)
843
+ response = self.create_error_response(str(e))
844
+
845
+ if isinstance(response, ErrorResponse):
846
+ # If the request has failed, update the status to "failed".
847
+ response_id = request.request_id
848
+ async with self.response_store_lock:
849
+ stored_response = self.response_store.get(response_id)
850
+ assert stored_response is not None
851
+ if stored_response.status not in ("completed", "cancelled"):
852
+ stored_response.status = "failed"
853
+
854
+ async def responses_background_stream_generator(
855
+ self,
856
+ response_id: str,
857
+ starting_after: Optional[int] = None,
858
+ ):
859
+ if response_id not in self.event_store:
860
+ raise ValueError(f"Unknown response_id: {response_id}")
861
+
862
+ event_deque, new_event_signal = self.event_store[response_id]
863
+ start_index = 0 if starting_after is None else starting_after + 1
864
+ current_index = start_index
865
+
866
+ while True:
867
+ new_event_signal.clear()
868
+
869
+ # Yield existing events from start_index
870
+ while current_index < len(event_deque):
871
+ event = event_deque[current_index]
872
+ if event == "__STREAM_END__":
873
+ return
874
+ yield event
875
+ current_index += 1
876
+
877
+ await new_event_signal.wait()
878
+
879
+ async def retrieve_responses(
880
+ self,
881
+ response_id: str,
882
+ starting_after: Optional[int],
883
+ stream: Optional[bool],
884
+ ) -> Union[ErrorResponse, ResponsesResponse]:
885
+ if not response_id.startswith("resp_"):
886
+ return self._make_invalid_id_error(response_id)
887
+
888
+ async with self.response_store_lock:
889
+ response = self.response_store.get(response_id)
890
+
891
+ if response is None:
892
+ return self._make_not_found_error(response_id)
893
+
894
+ if stream:
895
+ return self.responses_background_stream_generator(
896
+ response_id,
897
+ starting_after,
898
+ )
899
+ return response
900
+
901
+ async def cancel_responses(
902
+ self,
903
+ response_id: str,
904
+ ) -> Union[ErrorResponse, ResponsesResponse]:
905
+ if not response_id.startswith("resp_"):
906
+ return self._make_invalid_id_error(response_id)
907
+
908
+ async with self.response_store_lock:
909
+ response = self.response_store.get(response_id)
910
+ if response is None:
911
+ return self._make_not_found_error(response_id)
912
+
913
+ prev_status = response.status
914
+ if prev_status not in ("queued", "in_progress"):
915
+ return self.create_error_response(
916
+ err_type="invalid_request_error",
917
+ message="Cannot cancel a synchronous response.",
918
+ )
919
+
920
+ # Update the status to "cancelled".
921
+ response.status = "cancelled"
922
+
923
+ # Abort the request.
924
+ if (task := self.background_tasks.get(response_id)):
925
+ task.cancel()
926
+ try:
927
+ await task
928
+ except asyncio.CancelledError:
929
+ logger.exception("Background task for %s was cancelled",
930
+ response_id)
931
+ return response
932
+
933
+ def _make_invalid_id_error(self, response_id: str) -> ErrorResponse:
934
+ return self.create_error_response(
935
+ err_type="invalid_request_error",
936
+ message=(f"Invalid 'response_id': '{response_id}'. "
937
+ "Expected an ID that begins with 'resp'."),
938
+ )
939
+
940
+ def _make_not_found_error(self, response_id: str) -> ErrorResponse:
941
+ return self.create_error_response(
942
+ err_type="invalid_request_error",
943
+ message=f"Response with id '{response_id}' not found.",
944
+ status_code=HTTPStatus.NOT_FOUND,
945
+ )
946
+
947
+ def _make_store_not_supported_error(self) -> ErrorResponse:
948
+ return self.create_error_response(
949
+ err_type="invalid_request_error",
950
+ message=("`store=True` (default) is not supported. Please set "
951
+ "`store=False` in Responses API or set "
952
+ "`VLLM_ENABLE_RESPONSES_API_STORE=1` in the env var when "
953
+ "starting the vLLM server."),
954
+ status_code=HTTPStatus.BAD_REQUEST,
955
+ )
956
+
957
+ async def _process_simple_streaming_events(
958
+ self,
959
+ request: ResponsesRequest,
960
+ sampling_params: SamplingParams,
961
+ result_generator: AsyncIterator[Optional[ConversationContext]],
962
+ context: ConversationContext,
963
+ model_name: str,
964
+ tokenizer: AnyTokenizer,
965
+ request_metadata: RequestResponseMetadata,
966
+ created_time: int,
967
+ _send_event: Callable[[BaseModel], str],
968
+ ) -> AsyncGenerator[str, None]:
969
+ current_content_index = 0
970
+ current_output_index = 0
971
+ current_item_id = ""
972
+ reasoning_parser = None
973
+ if self.reasoning_parser:
974
+ reasoning_parser = self.reasoning_parser(tokenizer)
975
+ previous_text = ""
976
+ previous_token_ids: list[int] = []
977
+ first_delta_sent = False
978
+ previous_delta_messages: list[DeltaMessage] = []
979
+ async for ctx in result_generator:
980
+ assert isinstance(ctx, SimpleContext)
981
+ if ctx.last_output is None:
982
+ continue
983
+ if ctx.last_output.outputs:
984
+ output = ctx.last_output.outputs[0]
985
+ if reasoning_parser:
986
+ delta_message = \
987
+ reasoning_parser.extract_reasoning_content_streaming(
988
+ previous_text=previous_text,
989
+ current_text=previous_text + output.text,
990
+ delta_text=output.text,
991
+ previous_token_ids=previous_token_ids,
992
+ current_token_ids=previous_token_ids +
993
+ output.token_ids,
994
+ delta_token_ids=output.token_ids,
995
+ )
996
+ else:
997
+ delta_message = DeltaMessage(content=output.text, )
998
+ previous_text += output.text
999
+ previous_token_ids += output.token_ids
1000
+ if not delta_message:
1001
+ continue
1002
+ if not first_delta_sent:
1003
+ current_item_id = str(uuid.uuid4())
1004
+ if delta_message.reasoning_content:
1005
+ yield _send_event(
1006
+ openai_responses_types.
1007
+ ResponseOutputItemAddedEvent(
1008
+ type="response.output_item.added",
1009
+ sequence_number=-1,
1010
+ output_index=current_output_index,
1011
+ item=openai_responses_types.
1012
+ ResponseReasoningItem(
1013
+ type="reasoning",
1014
+ id=current_item_id,
1015
+ summary=[],
1016
+ status="in_progress",
1017
+ ),
1018
+ ))
1019
+ else:
1020
+ yield _send_event(
1021
+ openai_responses_types.
1022
+ ResponseOutputItemAddedEvent(
1023
+ type="response.output_item.added",
1024
+ sequence_number=-1,
1025
+ output_index=current_output_index,
1026
+ item=openai_responses_types.
1027
+ ResponseOutputMessage(
1028
+ id=current_item_id,
1029
+ type="message",
1030
+ role="assistant",
1031
+ content=[],
1032
+ status="in_progress",
1033
+ ),
1034
+ ))
1035
+ yield _send_event(
1036
+ openai_responses_types.ResponseContentPartAddedEvent(
1037
+ type="response.content_part.added",
1038
+ sequence_number=-1,
1039
+ output_index=current_output_index,
1040
+ item_id=current_item_id,
1041
+ content_index=current_content_index,
1042
+ part=openai_responses_types.ResponseOutputText(
1043
+ type="output_text",
1044
+ text="",
1045
+ annotations=[],
1046
+ logprobs=[],
1047
+ ),
1048
+ ))
1049
+ current_content_index += 1
1050
+ first_delta_sent = True
1051
+ # todo(kebe7jun) tool call support
1052
+
1053
+ # check delta message and previous delta message are
1054
+ # same as content or reasoning content
1055
+ if (previous_delta_messages
1056
+ and previous_delta_messages[-1].reasoning_content
1057
+ is not None and delta_message.content is not None):
1058
+ # from reasoning to normal content, send done
1059
+ # event for reasoning
1060
+ reason_content = ''.join(
1061
+ pm.reasoning_content for pm in previous_delta_messages
1062
+ if pm.reasoning_content is not None)
1063
+ yield _send_event(
1064
+ ResponseReasoningTextDoneEvent(
1065
+ type="response.reasoning_text.done",
1066
+ item_id=current_item_id,
1067
+ sequence_number=-1,
1068
+ output_index=current_output_index,
1069
+ content_index=current_content_index,
1070
+ text=reason_content,
1071
+ ))
1072
+ current_content_index = 0
1073
+ reasoning_item = ResponseReasoningItem(
1074
+ type="reasoning",
1075
+ content=[
1076
+ ResponseReasoningTextContent(
1077
+ text=reason_content,
1078
+ type="reasoning_text",
1079
+ ),
1080
+ ],
1081
+ status="completed",
1082
+ id=current_item_id,
1083
+ summary=[],
1084
+ )
1085
+ yield _send_event(
1086
+ ResponseOutputItemDoneEvent(
1087
+ type="response.output_item.done",
1088
+ sequence_number=-1,
1089
+ output_index=current_output_index,
1090
+ item=reasoning_item,
1091
+ ))
1092
+ yield _send_event(
1093
+ openai_responses_types.ResponseOutputItemAddedEvent(
1094
+ type="response.output_item.added",
1095
+ sequence_number=-1,
1096
+ output_index=current_output_index,
1097
+ item=openai_responses_types.ResponseOutputMessage(
1098
+ id=current_item_id,
1099
+ type="message",
1100
+ role="assistant",
1101
+ content=[],
1102
+ status="in_progress",
1103
+ ),
1104
+ ))
1105
+ current_output_index += 1
1106
+ current_item_id = str(uuid.uuid4())
1107
+ yield _send_event(
1108
+ openai_responses_types.ResponseContentPartAddedEvent(
1109
+ type="response.content_part.added",
1110
+ sequence_number=-1,
1111
+ output_index=current_output_index,
1112
+ item_id=current_item_id,
1113
+ content_index=current_content_index,
1114
+ part=openai_responses_types.ResponseOutputText(
1115
+ type="output_text",
1116
+ text="",
1117
+ annotations=[],
1118
+ logprobs=[],
1119
+ ),
1120
+ ))
1121
+ current_content_index += 1
1122
+ # reset previous delta messages
1123
+ previous_delta_messages = []
1124
+
1125
+ if delta_message.reasoning_content is not None:
1126
+ yield _send_event(
1127
+ ResponseReasoningTextDeltaEvent(
1128
+ type="response.reasoning_text.delta",
1129
+ sequence_number=-1,
1130
+ content_index=current_content_index,
1131
+ output_index=current_output_index,
1132
+ item_id=current_item_id,
1133
+ delta=delta_message.reasoning_content,
1134
+ ))
1135
+ elif delta_message.content is not None:
1136
+ yield _send_event(
1137
+ openai_responses_types.ResponseTextDeltaEvent(
1138
+ type="response.output_text.delta",
1139
+ sequence_number=-1,
1140
+ content_index=current_content_index,
1141
+ output_index=current_output_index,
1142
+ item_id=current_item_id,
1143
+ delta=delta_message.content,
1144
+ logprobs=self._create_stream_response_logprobs(
1145
+ token_ids=output.token_ids,
1146
+ logprobs=output.logprobs,
1147
+ tokenizer=tokenizer,
1148
+ top_logprobs=request.top_logprobs,
1149
+ ) if request.is_include_output_logprobs() else [],
1150
+ ))
1151
+ current_content_index += 1
1152
+
1153
+ previous_delta_messages.append(delta_message)
1154
+ if previous_delta_messages:
1155
+ if previous_delta_messages[-1].reasoning_content is not None:
1156
+ reason_content = ''.join(pm.reasoning_content
1157
+ for pm in previous_delta_messages
1158
+ if pm.reasoning_content is not None)
1159
+ yield _send_event(
1160
+ ResponseReasoningTextDoneEvent(
1161
+ type="response.reasoning_text.done",
1162
+ item_id=current_item_id,
1163
+ sequence_number=-1,
1164
+ output_index=current_output_index,
1165
+ content_index=current_content_index,
1166
+ text=reason_content,
1167
+ ))
1168
+ current_content_index += 1
1169
+ reasoning_item = ResponseReasoningItem(
1170
+ type="reasoning",
1171
+ content=[
1172
+ ResponseReasoningTextContent(
1173
+ text=reason_content,
1174
+ type="reasoning_text",
1175
+ ),
1176
+ ],
1177
+ status="completed",
1178
+ id=current_item_id,
1179
+ summary=[],
1180
+ )
1181
+ yield _send_event(
1182
+ ResponseOutputItemDoneEvent(
1183
+ type="response.output_item.done",
1184
+ sequence_number=-1,
1185
+ output_index=current_output_index,
1186
+ item=reasoning_item,
1187
+ ))
1188
+ elif previous_delta_messages[-1].content is not None:
1189
+ final_content = ''.join(pm.content
1190
+ for pm in previous_delta_messages
1191
+ if pm.content is not None)
1192
+ yield _send_event(
1193
+ openai_responses_types.ResponseTextDoneEvent(
1194
+ type="response.output_text.done",
1195
+ sequence_number=-1,
1196
+ output_index=current_output_index,
1197
+ content_index=current_content_index,
1198
+ text=final_content,
1199
+ logprobs=[],
1200
+ item_id=current_item_id,
1201
+ ))
1202
+ current_content_index += 1
1203
+ part = ResponseOutputText(
1204
+ text=final_content,
1205
+ type="output_text",
1206
+ annotations=[],
1207
+ )
1208
+ yield _send_event(
1209
+ openai_responses_types.ResponseContentPartDoneEvent(
1210
+ type="response.content_part.done",
1211
+ sequence_number=-1,
1212
+ item_id=current_item_id,
1213
+ output_index=current_output_index,
1214
+ content_index=current_content_index,
1215
+ part=part,
1216
+ ))
1217
+ current_content_index += 1
1218
+ item = ResponseOutputMessage(
1219
+ type="message",
1220
+ role="assistant",
1221
+ content=[
1222
+ part,
1223
+ ],
1224
+ status="completed",
1225
+ id=current_item_id,
1226
+ summary=[],
1227
+ )
1228
+ yield _send_event(
1229
+ ResponseOutputItemDoneEvent(
1230
+ type="response.output_item.done",
1231
+ sequence_number=-1,
1232
+ output_index=current_output_index,
1233
+ item=item,
1234
+ ))
1235
+
1236
+ async def _process_harmony_streaming_events(
1237
+ self,
1238
+ request: ResponsesRequest,
1239
+ sampling_params: SamplingParams,
1240
+ result_generator: AsyncIterator[Optional[ConversationContext]],
1241
+ context: ConversationContext,
1242
+ model_name: str,
1243
+ tokenizer: AnyTokenizer,
1244
+ request_metadata: RequestResponseMetadata,
1245
+ created_time: int,
1246
+ _send_event: Callable[[BaseModel], str],
1247
+ ) -> AsyncGenerator[str, None]:
1248
+ current_content_index = 0 # FIXME: this number is never changed
1249
+ current_output_index = 0
1250
+ current_item_id = "" # FIXME: this number is never changed
1251
+ sent_output_item_added = False
1252
+
1253
+ async for ctx in result_generator:
1254
+
1255
+ assert isinstance(ctx, StreamingHarmonyContext)
1256
+
1257
+ if ctx.is_expecting_start():
1258
+ current_output_index += 1
1259
+ sent_output_item_added = False
1260
+
1261
+ if len(ctx.parser.messages) > 0:
1262
+ previous_item = ctx.parser.messages[-1]
1263
+ if previous_item.recipient is not None:
1264
+ # Deal with tool call here
1265
+ pass
1266
+ elif previous_item.channel == "analysis":
1267
+ reasoning_item = ResponseReasoningItem(
1268
+ type="reasoning",
1269
+ content=[
1270
+ ResponseReasoningTextContent(
1271
+ text=previous_item.content[0].text,
1272
+ type="reasoning_text",
1273
+ ),
1274
+ ],
1275
+ status="completed",
1276
+ id=current_item_id,
1277
+ summary=[],
1278
+ )
1279
+ yield _send_event(
1280
+ ResponseReasoningTextDoneEvent(
1281
+ type="response.reasoning_text.done",
1282
+ item_id=current_item_id,
1283
+ sequence_number=-1,
1284
+ output_index=current_output_index,
1285
+ content_index=current_content_index,
1286
+ text=previous_item.content[0].text,
1287
+ ))
1288
+ yield _send_event(
1289
+ ResponseOutputItemDoneEvent(
1290
+ type="response.output_item.done",
1291
+ sequence_number=-1,
1292
+ output_index=current_output_index,
1293
+ item=reasoning_item,
1294
+ ))
1295
+ elif previous_item.channel == "final":
1296
+ text_content = ResponseOutputText(
1297
+ type="output_text",
1298
+ text=previous_item.content[0].text,
1299
+ annotations=[],
1300
+ )
1301
+ yield _send_event(
1302
+ openai_responses_types.ResponseTextDoneEvent(
1303
+ type="response.output_text.done",
1304
+ sequence_number=-1,
1305
+ output_index=current_output_index,
1306
+ content_index=current_content_index,
1307
+ text=previous_item.content[0].text,
1308
+ logprobs=[],
1309
+ item_id=current_item_id,
1310
+ ))
1311
+ yield _send_event(
1312
+ openai_responses_types.
1313
+ ResponseContentPartDoneEvent(
1314
+ type="response.content_part.done",
1315
+ sequence_number=-1,
1316
+ item_id=current_item_id,
1317
+ output_index=current_output_index,
1318
+ content_index=current_content_index,
1319
+ part=text_content,
1320
+ ))
1321
+ yield _send_event(
1322
+ openai_responses_types.ResponseOutputItemDoneEvent(
1323
+ type="response.output_item.done",
1324
+ sequence_number=-1,
1325
+ output_index=current_output_index,
1326
+ item=ResponseOutputMessage(
1327
+ id=current_item_id,
1328
+ type="message",
1329
+ role="assistant",
1330
+ content=[text_content],
1331
+ status="completed",
1332
+ ),
1333
+ ))
1334
+
1335
+ if ctx.parser.last_content_delta:
1336
+ if (ctx.parser.current_channel == "final"
1337
+ and ctx.parser.current_recipient is None):
1338
+ if not sent_output_item_added:
1339
+ sent_output_item_added = True
1340
+ yield _send_event(
1341
+ openai_responses_types.
1342
+ ResponseOutputItemAddedEvent(
1343
+ type="response.output_item.added",
1344
+ sequence_number=-1,
1345
+ output_index=current_output_index,
1346
+ item=openai_responses_types.
1347
+ ResponseOutputMessage(
1348
+ id=current_item_id,
1349
+ type="message",
1350
+ role="assistant",
1351
+ content=[],
1352
+ status="in_progress",
1353
+ ),
1354
+ ))
1355
+ yield _send_event(
1356
+ openai_responses_types.
1357
+ ResponseContentPartAddedEvent(
1358
+ type="response.content_part.added",
1359
+ sequence_number=-1,
1360
+ output_index=current_output_index,
1361
+ item_id=current_item_id,
1362
+ content_index=current_content_index,
1363
+ part=openai_responses_types.ResponseOutputText(
1364
+ type="output_text",
1365
+ text="",
1366
+ annotations=[],
1367
+ logprobs=[],
1368
+ ),
1369
+ ))
1370
+ yield _send_event(
1371
+ openai_responses_types.ResponseTextDeltaEvent(
1372
+ type="response.output_text.delta",
1373
+ sequence_number=-1,
1374
+ content_index=current_content_index,
1375
+ output_index=current_output_index,
1376
+ item_id=current_item_id,
1377
+ delta=ctx.parser.last_content_delta,
1378
+ # TODO, use logprobs from ctx.last_request_output
1379
+ logprobs=[],
1380
+ ))
1381
+ elif (ctx.parser.current_channel == "analysis"
1382
+ and ctx.parser.current_recipient is None):
1383
+ if not sent_output_item_added:
1384
+ sent_output_item_added = True
1385
+ yield _send_event(
1386
+ openai_responses_types.
1387
+ ResponseOutputItemAddedEvent(
1388
+ type="response.output_item.added",
1389
+ sequence_number=-1,
1390
+ output_index=current_output_index,
1391
+ item=openai_responses_types.
1392
+ ResponseReasoningItem(
1393
+ type="reasoning",
1394
+ id=current_item_id,
1395
+ summary=[],
1396
+ status="in_progress",
1397
+ ),
1398
+ ))
1399
+ yield _send_event(
1400
+ openai_responses_types.
1401
+ ResponseContentPartAddedEvent(
1402
+ type="response.content_part.added",
1403
+ sequence_number=-1,
1404
+ output_index=current_output_index,
1405
+ item_id=current_item_id,
1406
+ content_index=current_content_index,
1407
+ part=openai_responses_types.ResponseOutputText(
1408
+ type="output_text",
1409
+ text="",
1410
+ annotations=[],
1411
+ logprobs=[],
1412
+ ),
1413
+ ))
1414
+ yield _send_event(
1415
+ ResponseReasoningTextDeltaEvent(
1416
+ type="response.reasoning_text.delta",
1417
+ item_id=current_item_id,
1418
+ output_index=current_output_index,
1419
+ content_index=current_content_index,
1420
+ delta=ctx.parser.last_content_delta,
1421
+ sequence_number=-1,
1422
+ ))
1423
+ # built-in tools will be triggered on the analysis channel
1424
+ # However, occasionally built-in tools will
1425
+ # still be output to commentary.
1426
+ elif (ctx.parser.current_channel == "commentary"
1427
+ or ctx.parser.current_channel == "analysis"
1428
+ ) and ctx.parser.current_recipient == "python":
1429
+ if not sent_output_item_added:
1430
+ sent_output_item_added = True
1431
+ yield _send_event(
1432
+ openai_responses_types.
1433
+ ResponseOutputItemAddedEvent(
1434
+ type="response.output_item.added",
1435
+ sequence_number=-1,
1436
+ output_index=current_output_index,
1437
+ item=openai_responses_types.
1438
+ ResponseCodeInterpreterToolCallParam(
1439
+ type="code_interpreter_call",
1440
+ id=current_item_id,
1441
+ code=None,
1442
+ container_id="auto",
1443
+ outputs=None,
1444
+ status="in_progress",
1445
+ ),
1446
+ ))
1447
+ yield _send_event(
1448
+ openai_responses_types.
1449
+ ResponseCodeInterpreterCallInProgressEvent(
1450
+ type=
1451
+ "response.code_interpreter_call.in_progress",
1452
+ sequence_number=-1,
1453
+ output_index=current_output_index,
1454
+ item_id=current_item_id,
1455
+ ))
1456
+ yield _send_event(
1457
+ openai_responses_types.
1458
+ ResponseCodeInterpreterCallCodeDeltaEvent(
1459
+ type="response.code_interpreter_call_code.delta",
1460
+ sequence_number=-1,
1461
+ output_index=current_output_index,
1462
+ item_id=current_item_id,
1463
+ delta=ctx.parser.last_content_delta,
1464
+ ))
1465
+ if ctx.is_assistant_action_turn() and len(ctx.parser.messages) > 0:
1466
+ previous_item = ctx.parser.messages[-1]
1467
+ if (self.tool_server is not None
1468
+ and self.tool_server.has_tool("browser")
1469
+ and previous_item.recipient is not None
1470
+ and previous_item.recipient.startswith("browser.")):
1471
+ function_name = previous_item.recipient[len("browser."):]
1472
+ action = None
1473
+ parsed_args = json.loads(previous_item.content[0].text)
1474
+ if function_name == "search":
1475
+ action = (openai_responses_types.
1476
+ response_function_web_search.ActionSearch(
1477
+ type="search",
1478
+ query=parsed_args["query"],
1479
+ ))
1480
+ elif function_name == "open":
1481
+ action = (
1482
+ openai_responses_types.
1483
+ response_function_web_search.ActionOpenPage(
1484
+ type="open_page",
1485
+ # TODO: translate to url
1486
+ url=f"cursor:{parsed_args.get('cursor', '')}",
1487
+ ))
1488
+ elif function_name == "find":
1489
+ action = (
1490
+ openai_responses_types.
1491
+ response_function_web_search.ActionFind(
1492
+ type="find",
1493
+ pattern=parsed_args["pattern"],
1494
+ # TODO: translate to url
1495
+ url=f"cursor:{parsed_args.get('cursor', '')}",
1496
+ ))
1497
+ else:
1498
+ raise ValueError(
1499
+ f"Unknown function name: {function_name}")
1500
+
1501
+ yield _send_event(
1502
+ openai_responses_types.ResponseOutputItemAddedEvent(
1503
+ type="response.output_item.added",
1504
+ sequence_number=-1,
1505
+ output_index=current_output_index,
1506
+ item=openai_responses_types.
1507
+ response_function_web_search.
1508
+ ResponseFunctionWebSearch(
1509
+ # TODO: generate a unique id for web search call
1510
+ type="web_search_call",
1511
+ id=current_item_id,
1512
+ action=action,
1513
+ status="in_progress",
1514
+ ),
1515
+ ))
1516
+ yield _send_event(
1517
+ openai_responses_types.
1518
+ ResponseWebSearchCallInProgressEvent(
1519
+ type="response.web_search_call.in_progress",
1520
+ sequence_number=-1,
1521
+ output_index=current_output_index,
1522
+ item_id=current_item_id,
1523
+ ))
1524
+ yield _send_event(
1525
+ openai_responses_types.
1526
+ ResponseWebSearchCallSearchingEvent(
1527
+ type="response.web_search_call.searching",
1528
+ sequence_number=-1,
1529
+ output_index=current_output_index,
1530
+ item_id=current_item_id,
1531
+ ))
1532
+
1533
+ # enqueue
1534
+ yield _send_event(
1535
+ openai_responses_types.
1536
+ ResponseWebSearchCallCompletedEvent(
1537
+ type="response.web_search_call.completed",
1538
+ sequence_number=-1,
1539
+ output_index=current_output_index,
1540
+ item_id=current_item_id,
1541
+ ))
1542
+ yield _send_event(
1543
+ openai_responses_types.ResponseOutputItemDoneEvent(
1544
+ type="response.output_item.done",
1545
+ sequence_number=-1,
1546
+ output_index=current_output_index,
1547
+ item=openai_responses_types.
1548
+ ResponseFunctionWebSearch(
1549
+ type="web_search_call",
1550
+ id=current_item_id,
1551
+ action=action,
1552
+ status="completed",
1553
+ ),
1554
+ ))
1555
+
1556
+ if (self.tool_server is not None
1557
+ and self.tool_server.has_tool("python")
1558
+ and previous_item.recipient is not None
1559
+ and previous_item.recipient.startswith("python")):
1560
+ yield _send_event(
1561
+ openai_responses_types.
1562
+ ResponseCodeInterpreterCallCodeDoneEvent(
1563
+ type="response.code_interpreter_call_code.done",
1564
+ sequence_number=-1,
1565
+ output_index=current_output_index,
1566
+ item_id=current_item_id,
1567
+ code=previous_item.content[0].text,
1568
+ ))
1569
+ yield _send_event(
1570
+ openai_responses_types.
1571
+ ResponseCodeInterpreterCallInterpretingEvent(
1572
+ type="response.code_interpreter_call.interpreting",
1573
+ sequence_number=-1,
1574
+ output_index=current_output_index,
1575
+ item_id=current_item_id,
1576
+ ))
1577
+ yield _send_event(
1578
+ openai_responses_types.
1579
+ ResponseCodeInterpreterCallCompletedEvent(
1580
+ type="response.code_interpreter_call.completed",
1581
+ sequence_number=-1,
1582
+ output_index=current_output_index,
1583
+ item_id=current_item_id,
1584
+ ))
1585
+ yield _send_event(
1586
+ openai_responses_types.ResponseOutputItemDoneEvent(
1587
+ type="response.output_item.done",
1588
+ sequence_number=-1,
1589
+ output_index=current_output_index,
1590
+ item=openai_responses_types.
1591
+ ResponseCodeInterpreterToolCallParam(
1592
+ type="code_interpreter_call",
1593
+ id=current_item_id,
1594
+ code=previous_item.content[0].text,
1595
+ container_id="auto",
1596
+ # TODO: add outputs here
1597
+ outputs=[],
1598
+ status="completed",
1599
+ ),
1600
+ ))
1601
+
1602
+ async def responses_stream_generator(
1603
+ self,
1604
+ request: ResponsesRequest,
1605
+ sampling_params: SamplingParams,
1606
+ result_generator: AsyncIterator[Optional[ConversationContext]],
1607
+ context: ConversationContext,
1608
+ model_name: str,
1609
+ tokenizer: AnyTokenizer,
1610
+ request_metadata: RequestResponseMetadata,
1611
+ created_time: Optional[int] = None,
1612
+ ) -> AsyncGenerator[str, None]:
1613
+ # TODO:
1614
+ # 1. Handle disconnect
1615
+
1616
+ created_time = created_time or int(time.time())
1617
+
1618
+ sequence_number = 0
1619
+
1620
+ def _send_event(event: BaseModel):
1621
+ nonlocal sequence_number
1622
+ # Set sequence_number if the event has this attribute
1623
+ if hasattr(event, 'sequence_number'):
1624
+ event.sequence_number = sequence_number
1625
+ sequence_number += 1
1626
+ # Get event type from the event's type field if it exists
1627
+ event_type = getattr(event, 'type', 'unknown')
1628
+ return (f"event: {event_type}\n"
1629
+ f"data: {event.model_dump_json(indent=None)}\n\n")
1630
+
1631
+ async with AsyncExitStack() as exit_stack:
1632
+ processer = None
1633
+ if self.use_harmony:
1634
+ await context.init_tool_sessions(self.tool_server, exit_stack,
1635
+ request.request_id)
1636
+ processer = self._process_harmony_streaming_events
1637
+ else:
1638
+ processer = self._process_simple_streaming_events
1639
+
1640
+ initial_response = ResponsesResponse.from_request(
1641
+ request,
1642
+ sampling_params,
1643
+ model_name=model_name,
1644
+ created_time=created_time,
1645
+ output=[],
1646
+ status="in_progress",
1647
+ usage=None,
1648
+ ).model_dump()
1649
+ yield _send_event(
1650
+ ResponseCreatedEvent(
1651
+ type="response.created",
1652
+ sequence_number=-1,
1653
+ response=initial_response,
1654
+ ))
1655
+ yield _send_event(
1656
+ ResponseInProgressEvent(
1657
+ type="response.in_progress",
1658
+ sequence_number=-1,
1659
+ response=initial_response,
1660
+ ))
1661
+
1662
+ async for event_data in processer(request, sampling_params,
1663
+ result_generator, context,
1664
+ model_name, tokenizer,
1665
+ request_metadata, created_time,
1666
+ _send_event):
1667
+ yield event_data
1668
+
1669
+ async def empty_async_generator():
1670
+ # A hack to trick Python to think this is a generator but
1671
+ # in fact it immediately returns.
1672
+ if False:
1673
+ yield
1674
+
1675
+ final_response = await self.responses_full_generator(
1676
+ request,
1677
+ sampling_params,
1678
+ empty_async_generator(),
1679
+ context,
1680
+ model_name,
1681
+ tokenizer,
1682
+ request_metadata,
1683
+ created_time=created_time,
1684
+ )
1685
+ yield _send_event(
1686
+ openai_responses_types.ResponseCompletedEvent(
1687
+ type="response.completed",
1688
+ sequence_number=-1,
1689
+ response=final_response.model_dump(),
1690
+ ))