3 people b730c46352 [Perf] [Hybrid] Fused Triton kernel for GPU-side Mamba state postprocessing (#40172) ...

Signed-off-by: Francesco Fusco <ffu@zurich.ibm.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

2026-05-21 04:50:54 -07:00

..

__init__.py

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

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

test_gpu_input_batch.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_gpu_model_runner_v2_eplb.py

[Feature] EPLB Support for GPU Model Runner v2 (#37488)

2026-03-25 08:16:39 -07:00

test_gpu_model_runner.py

[Bugfix][Spec Decode] Wire draft_probs into probabilistic draft_model rejection (#40269)

2026-05-13 21:04:03 -04:00

test_gpu_profiler.py

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

2026-01-22 09:45:40 -08:00

test_late_interaction_runner.py

[Frontend] Re-enable running MaxSim on GPU (#38620)

2026-04-03 00:03:13 +08:00

test_mamba_utils.py

[Perf] [Hybrid] Fused Triton kernel for GPU-side Mamba state postprocessing (#40172)

2026-05-21 04:50:54 -07: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