Giancarlo Delfin 5daf62271d [Model Runner V2] Fuse probabilistic rejection sample kernels (#38496) ...

Signed-off-by: Giancarlo Delfin <gdelfin@inferact.ai>

2026-04-07 17:37:37 -07:00

..

__init__.py

Test: added acceptance length tests (#32030)

2026-01-20 18:55:15 +00:00

test_acceptance_length.py

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

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

test_backup_token_async_spec.py

[Bugfix][Async] Fix async spec decoding with hybrid models (#38556)

2026-03-31 11:08:54 -04:00

test_eagle_step_kernel.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_eagle.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_extract_hidden_states.py

[Bugfix][Spec Decode] Fix extract_hidden_states for VLM models (#38987)

2026-04-05 02:41:54 -07:00

test_max_len.py

[Attention] Update tests to remove deprecated env vars (#30563)

2025-12-17 09:49:59 -08:00

test_mtp.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_ngram.py

[Cleanup] Remove obsolete spec decoding compatibility logic (#32003)

2026-01-09 05:44:18 +00:00

test_probabilistic_rejection_sampler_utils.py

[Model Runner V2] Fuse probabilistic rejection sample kernels (#38496)

2026-04-07 17:37:37 -07:00

test_speculators_eagle3.py

[Rocm][CI] Fix test_speculator_eagle3 by skipping the CompressedTensorw4a16 Model (#30001)

2025-12-04 07:52:28 +00:00

test_synthetic_rejection_sampler_utils.py

[Model Runner V2] Enable forcing a specific acceptance rate during rejection sampling (#38045)

2026-03-26 13:38:12 -07:00

test_tree_attention.py

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

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