vLLM/tests/v1/worker at 09fddeb4ce7ee8cd0c46f1da9c45aafb9a17e89a - vLLM

__init__.py

[V1] Adding min tokens/repetition/presence/frequence penalties to V1 sampler (#10681 )

2024-12-26 19:02:58 +09:00

test_attn_utils.py

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718 )

2026-08-21 17:47:49 -07:00

test_cp_utils.py

DCP supports hybrid attention (#40996 )

2026-07-09 21:34:45 -07:00

test_dsv4_packed_zeroer_geometry.py

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718 )

2026-08-21 17:47:49 -07:00

test_encoder_runner.py

[ModelRunnerV2] Support prompt embeds (#42963 )

2026-08-17 21:58:50 +00:00

test_gpu_autoregressive_speculator.py

[CI][Bugfix][V1] Remove stale FlashAttention metadata arguments (#51854 )

2026-08-11 18:54:56 +00:00

test_gpu_bad_words.py

[Bugfix][Model Runner V2][Spec Decode] Fix off-by-one in bad_words draft-prefix matching (#52311 )

2026-08-16 09:02:20 +00:00

test_gpu_batch_ordering.py

[Spec Decode] DSpark confidence-scheduled verification (#47808 )

2026-08-12 12:27:48 -07:00

test_gpu_batch_shard.py

[Model Runner V2] batch-sharded sample (#50465 )

2026-08-24 15:49:57 -07:00

test_gpu_block_table.py

[Bugfix][Kernel] Handle kernel block sizes in V2 DCP slot mapping (#51031 )

2026-08-24 07:28:38 +00:00

test_gpu_extract_hidden_states_speculator.py

[Feature][Model Runner V2] Support extract_hidden_states speculation (#49811 )

2026-08-20 11:13:43 -04:00

test_gpu_gumbel_sample.py

[Model Runner V2][Spec Decode] Fix draft logits cache column stride in gumbel_sample (#53017 )

2026-08-19 17:39:30 -07:00

test_gpu_input_batch_v2.py

[XPU][CI] Use platform device in InputBatch V2 test (#49939 )

2026-07-27 15:28:02 +08:00

test_gpu_input_batch.py

[Pooling] Improve BGE-M3 sync pooling throughput by up to 3.13% (#53464 )

2026-08-24 08:33:32 -04:00

test_gpu_model_runner_mm_gather.py

Encoder cache extension hooks (#48218 )

2026-07-24 02:52:00 -07:00

test_gpu_model_runner_v2_cudagraph_profiling.py

[Model Runner V2] Reserve CUDA graph memory (#53306 )

2026-08-23 23:11:19 -07:00

test_gpu_model_runner_v2_eplb.py

[Model Runner V2] batch-sharded sample (#50465 )

2026-08-24 15:49:57 -07:00

test_gpu_model_runner_v2.py

[Kimi-K3] support DCP partial prefix cache hit (#50493 )

2026-08-17 17:05:51 -07:00

test_gpu_model_runner.py

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718 )

2026-08-21 17:47:49 -07:00

test_gpu_profiler.py

[Profiler] Add minimal Triton Proton profiling backend (#48789 )

2026-08-11 14:51:57 -07:00

test_gpu_rejection_sampler_chunking.py

[MRV2][Spec Decode] Avoid rejection sampler OOM by chunking (#48630 )

2026-07-23 18:13:13 +08:00

test_gpu_rejection_sampler_i64.py

[Bugfix][Model Runner V2][Spec Decode] Fix int32 offset overflow in block verification kernels (#47383 )

2026-07-02 07:32:38 -07:00

test_gpu_sampler_flags.py

[Performance][MRV2] Cache logits-processing request state (#52329 )

2026-08-17 02:36:41 +00:00

test_gpu_thinking_budget.py

[Performance][MRV2] Cache logits-processing request state (#52329 )

2026-08-17 02:36:41 +00:00

test_gpu_trace_replay.py

[Core][V1] Support trace_decode_token_ids for deterministic decode replay (#46701 )

2026-08-20 15:12:43 +08:00

test_gpu_warmup_blocks.py

[Refactor][Model Runner V2][Multimodal] Move the encoder-only path out of the shared runner (#53176 )

2026-08-22 05:11:44 +00:00

test_gpu_worker_weight_transfer.py

[Bugfix] Fix level-2 sleep/wake/reload with enable_lora=True (#39935 )

2026-08-06 09:13:10 +08:00

test_gpu_worker.py

[Misc] Move PyNvVideoCodec stuff out of gpu worker (#49322 )

2026-07-22 10:11:56 +08:00

test_jit_warmup_migration.py

[3/N][Feat][Perf] Add new warmup infrastructure for JITs. Add provider registry and orchestration for JIT warmup (#50174 )

2026-08-17 10:11:14 -07:00

test_kv_block_zeroer.py

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718 )

2026-08-21 17:47:49 -07:00

test_late_interaction_runner.py

[Core] Move MRV1 late_interaction_runner.py out of MRV2 subtree (#48014 )

2026-07-08 18:30:11 +00:00

test_mamba_hybrid_model_state.py

[K3] support recoverssm for K3 (#51855 )

2026-08-17 19:34:14 +08:00

test_mamba_utils.py

[Bugfix][XPU] Fix Mamba state pointer overflow (#48109 )

2026-08-19 10:45:41 +08:00

test_mixed_warmup_gate.py

[BugFix] Gate MRV2 mixed sparse-MLA warmup on max_num_seqs > 1 (#47050 )

2026-06-30 16:31:27 +01:00

test_mrope_prompt_embeds.py

[Security] Fix DoS via prompt_embeds on M-RoPE models (#45252 )

2026-06-13 10:17:38 +00:00

test_prompt_embeds_state.py

[ModelRunnerV2] Support prompt embeds (#42963 )

2026-08-17 21:58:50 +00:00

test_sleep_mode_backend.py

[Core] Make sleep-mode backend capability flags communicator-agnostic (#47243 )

2026-07-01 07:17:44 +00:00

test_utils.py

[V0 Deprecation] Refactor kv cache from list to element (#37487 )

2026-03-23 20:10:11 -07:00

test_worker_memory_snapshot.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145 )

2026-03-12 07:57:47 -07:00

test_workspace.py

[Bugfix] Make DSV4 sparse MLA work end-to-end for plain decode, MTP, and DSpark (#51538 )

2026-08-15 07:01:32 -07:00

test_xpu_model_runner.py

[XPU] Fix CUDA API shims breaking Torch Dynamo during AOT compile (#43092 )

2026-07-06 08:29:20 +08:00