Benjamin ChislettandWentao Ye 1cd75b3dd4 [Bugfix] Fix race condition in KVBlockZeroer (#48085) ...

Signed-off-by: Benjamin Chislett <bchislett@nvidia.com>
Signed-off-by: Benjamin Chislett <chislett.ben@gmail.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>

2026-07-09 09:18:19 +00:00

..

__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

[Spec Decode] Support mixed KV page sizes for DFlash (#45181)

2026-06-21 22:45:14 +08:00

test_encoder_runner.py

[ModelRunner V2] Support realtime embeddings (#46762)

2026-06-26 19:42:27 -07:00

test_gpu_autoregressive_speculator.py

[Model Runner V2][Spec Decode] Handle tuple hidden states from MTP draft models (#46786)

2026-06-27 18:38:07 -07:00

test_gpu_block_table.py

[PERF] Fuse multi-group block table staged writes (#44944)

2026-06-16 10:53:27 -07:00

test_gpu_gumbel_sample.py

[MRV2] Make FP32 Gumbel sampling more accurate (#45996)

2026-06-18 19:42:09 +00:00

test_gpu_input_batch.py

[Bugfix] Prevent padding placeholders from reaching embeddings (#47029)

2026-07-01 09:26:03 -07:00

test_gpu_model_runner_mm_gather.py

[Bugfix][Spec Decode] Fix EAGLE drafter multimodal encoder cache misses (#46315)

2026-06-22 18:14:02 +00:00

test_gpu_model_runner_v2_eplb.py

[Model] Add DiffusionGemma Support (#45163)

2026-06-11 22:17:35 -07:00

test_gpu_model_runner.py

[Core][Engine] only materialize tokens when thinking budget is in req (#47053)

2026-07-07 21:02:38 -06:00

test_gpu_profiler.py

Support custom URI schemes and trace handlers for profiler (#32393)

2026-01-22 09:45:40 -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_worker_weight_transfer.py

Weight sync refactor + move sparse nccl engine (#44353)

2026-07-01 01:25:19 -07:00

test_gpu_worker.py

[Core] Persist and reuse the memory-profiling result across boots (opt-in) (#47388)

2026-07-08 00:53:02 +00:00

test_kv_block_zeroer.py

[Bugfix] Fix race condition in KVBlockZeroer (#48085)

2026-07-09 09:18:19 +00: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_utils.py

[Kernel] Support DS Mamba tail copy for MTP align mode (#45473)

2026-06-16 22:50:30 +00: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_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_xpu_model_runner.py

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

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