hf_text-generation-inference/server/text_generation_server/layers/moe
Daniël de Kok 64142489b6
Add support for fused MoE Marlin for AWQ (#2616)
* Add support for fused MoE Marlin for AWQ

This uses the updated MoE Marlin kernels from vLLM.

* Add integration test for AWQ MoE
2024-10-08 11:56:41 +02:00
..
__init__.py Add support for fused MoE Marlin for AWQ (#2616) 2024-10-08 11:56:41 +02:00
fused_moe_rocm.py Update ROCM libs and improvements (#2579) 2024-09-30 10:54:32 +02:00
gptq_marlin.py Add support for fused MoE Marlin for AWQ (#2616) 2024-10-08 11:56:41 +02:00
unquantized.py Update ROCM libs and improvements (#2579) 2024-09-30 10:54:32 +02:00