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

Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
Signed-off-by: Kunshang Ji <jikunshang95@gmail.com>

2026-06-30 14:39:52 +08: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_logprobs_e2e.py

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

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

test_logprobs.py

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

2026-06-30 14:39:52 +08: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