3 people 7aab6e2684 [ROCm][Bugfix] Enable the fp32 head_dtype torch.mm fast path on ROCm (#48688) ...

Signed-off-by: Turner Jabbour <doubleujabbour@gmail.com>
Co-authored-by: Claude <noreply@anthropic.com>

2026-07-15 08:18:22 +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_batched_count_greater_than.py

[Performance Improvement] Update batched_count_greater_than to handle batch size 1 without recompile (#38933)

2026-04-09 23:51:31 +08:00

test_head_dtype.py

[ROCm][Bugfix] Enable the fp32 head_dtype torch.mm fast path on ROCm (#48688)

2026-07-15 08:18:22 +00:00

test_logprobs_e2e.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

test_logprobs.py

[Bugfix] Reject sampling params unsupported by diffusion models (#45418)

2026-07-07 11:25:36 +00:00

test_rejection_sampler.py

[Spec Decode] Reject placeholder (-1) draft tokens in rejection sampler (#46533)

2026-06-24 03:32:32 +00:00

test_sampler.py

refactor hard coded device string in test files under tests/v1 and tests/lora (#37566)

2026-04-03 11:21:47 +08:00

test_sampling_params_e2e.py

[Bugfix][Model Runner V2] Preserve all allowed_token_ids in the logit bias kernel (#46245)

2026-06-23 07:01:13 +00:00

test_topk_topp_sampler.py

[Platform] Replace torch.cuda.mem_get_info with torch.accelerator.get_memory_info (#44825)

2026-06-30 14:39:52 +08:00

utils.py

[Reasoning] Support for speculative decoding with thinking budget (#34668)

2026-04-29 06:14:52 +00:00