sroy745 f3a507f1d3 [Core] Add an environment variable which needs to be set explicitly to allow BlockSpaceManagerV1 (#9149)

2024-10-10 14:17:17 +08:00

..

e2e

[Core] Add an environment variable which needs to be set explicitly to allow BlockSpaceManagerV1 (#9149)

2024-10-10 14:17:17 +08:00

__init__.py

…

test_batch_expansion.py

[BugFix] Fix use of per-request seed with pipeline parallel (#6698)

2024-07-30 10:40:08 -07:00

test_dynamic_spec_decode.py

[Spec Decode] Disable Log Prob serialization to CPU for spec decoding for both draft and target models. (#6485)

2024-07-20 23:58:58 -07:00

test_metrics.py

[Bugfix] [SpecDecode] AsyncMetricsCollector: update time since last collection (#6578)

2024-07-19 14:01:03 -07:00

test_multi_step_worker.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

test_ngram_worker.py

…

test_scorer.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

test_spec_decode_worker.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

test_utils.py

[Bugfix] Unify rank computation across regular decoding and speculative decoding (#7899)

2024-08-28 22:18:13 -07:00

utils.py

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00