huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-21 23:12:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
de19e7e844
text-generation-inference/server/text_generation_server/layers/moe
History
Mohit Sharma 8f66d323d0
Update vllm kernels for ROCM (#2826) 
* (vllm) updated vllm rocm kernels

* revert silu

* update partition size

* remove grouped_topk

* (nit) remove log

* update moe-kernels commit
2024-12-18 12:44:42 +01:00
..
__init__.py Update vllm kernels for ROCM (#2826) 2024-12-18 12:44:42 +01:00
gptq_marlin.py Add support for fused MoE Marlin for AWQ (#2616) 2024-10-08 11:56:41 +02:00
unquantized.py Update vllm kernels for ROCM (#2826) 2024-12-18 12:44:42 +01:00