Divakar Verma c3dde367f1 [Kernel][ROCm][AMD] fused_moe Triton configs v2 for mi300X (#5932)

2024-06-27 13:41:08 -07:00

..

fused_moe

[Kernel][ROCm][AMD] fused_moe Triton configs v2 for mi300X (#5932)

2024-06-27 13:41:08 -07:00

ops

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00

quantization

[Misc] Update w4a16 compressed-tensors support to include w8a16 (#5794)

2024-06-25 19:23:35 +00:00

__init__.py

Change the name to vLLM (#150)

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

activation.py

[Kernel][CPU] Add Quick gelu to CPU (#5717)

2024-06-21 06:39:40 +00:00

layernorm.py

[Model] Add Gemma 2 (#5908)

2024-06-27 13:33:56 -07:00

linear.py

[Misc] Add channel-wise quantization support for w8a8 dynamic per token activation quantization (#5542)

2024-06-18 12:45:05 -04:00

logits_processor.py

[Model] Add Gemma 2 (#5908)

2024-06-27 13:33:56 -07:00

pooler.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

rejection_sampler.py

[Speculative Decoding 1/2 ] Add typical acceptance sampling as one of the sampling techniques in the verifier (#5131)

2024-06-17 21:29:09 -05:00

rotary_embedding.py

[Model] Add Gemma 2 (#5908)

2024-06-27 13:33:56 -07:00

sampler.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

spec_decode_base_sampler.py

[Speculative Decoding 1/2 ] Add typical acceptance sampling as one of the sampling techniques in the verifier (#5131)

2024-06-17 21:29:09 -05:00

typical_acceptance_sampler.py

[Speculative Decoding 1/2 ] Add typical acceptance sampling as one of the sampling techniques in the verifier (#5131)

2024-06-17 21:29:09 -05:00

vocab_parallel_embedding.py

[Bugfix] Fix embedding to support 2D inputs (#5829)

2024-06-26 00:15:22 -07:00