youkaichao 614aa51203 [misc][cuda] use nvml to avoid accidentally cuda initialization (#6007)

2024-06-30 20:07:34 -07:00

..

backends

[Kernel] Flashinfer for prefill & decode, with Cudagraph support for decode (#4628)

2024-06-28 15:28:49 -07:00

ops

[misc][cuda] use nvml to avoid accidentally cuda initialization (#6007)

2024-06-30 20:07:34 -07:00

__init__.py

[Core][2/N] Model runner refactoring part 2. Combine prepare prefill / decode to a single API (#4681)

2024-05-15 14:00:10 +09:00

layer.py

[Bugfix] Only add Attention.kv_scale if kv cache quantization is enabled (#5936)

2024-06-28 21:12:40 +00:00

selector.py

[Kernel] Flashinfer for prefill & decode, with Cudagraph support for decode (#4628)

2024-06-28 15:28:49 -07:00