shiyi.c_98 64da65b322 Prefix Caching- fix t4 triton error (#2517)

2024-02-16 14:17:55 -08:00

..

quantization

Refactor 2 awq gemm kernels into m16nXk32 (#2723)

2024-02-12 11:02:17 -08:00

triton_kernel

Prefix Caching- fix t4 triton error (#2517)

2024-02-16 14:17:55 -08:00

__init__.py

Change the name to vLLM (#150)

2023-06-17 03:07:40 -07:00

activation.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00

attention.py

[ROCm] support Radeon™ 7900 series (gfx1100) without using flash-attention (#2768)

2024-02-10 23:14:37 -08:00

fused_moe.py

Add fused top-K softmax kernel for MoE (#2769)

2024-02-05 17:38:02 -08:00

layernorm.py

Revert "Refactor llama family models (#2637)" (#2851)

2024-02-13 09:24:59 -08:00

linear.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00

rejection_sampler.py

[Speculative decoding 1/9] Optimized rejection sampler (#2336)

2024-01-09 15:38:41 -08:00

rotary_embedding.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00

sampler.py

[Experimental] Add multi-LoRA support (#1804)

2024-01-23 15:26:37 -08:00

vocab_parallel_embedding.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00