Kaiyang Chen 10c38e3e46

[Misc]: Implement CPU/GPU swapping in BlockManagerV2 (#3834)

2024-06-03 13:37:11 -07:00

..

block

[Misc]: Implement CPU/GPU swapping in BlockManagerV2 (#3834)

2024-06-03 13:37:11 -07:00

__init__.py

[Tests] Add block manager and scheduler tests (#3108)

2024-03-05 18:23:34 -08:00

test_block_manager.py

[Core] Avoid the need to pass None values to Sequence.inputs (#5099)

2024-05-29 16:05:01 -07:00

test_chunked_prefill_scheduler.py

[Core][Optimization] change python dict to pytorch tensor for blocks to swap (#4659)

2024-05-08 12:07:05 -07:00

test_scheduler.py

[Scheduler] Warning upon preemption and Swapping (#4647)

2024-05-13 23:50:44 +09:00

utils.py

[Core] Avoid the need to pass None values to Sequence.inputs (#5099)

2024-05-29 16:05:01 -07:00