跳到正文
原文
Hugging Face Blog·· 2025-07-09AI 评分36

Hugging Face 与 AMD 为 MI300X 打造自定义 kernel,优化 Llama 3.1 405B FP8 推理

Creating custom kernels for the AMD MI300

AI 导读

Hugging Face 与 AMD 合作,为 MI300X GPU 编写了三个开源自定义 kernel——融合残差连接/RMS norm/FP8 转换、融合 SwiGLU 激活/FP8 转换以及 Skinny GEMM,用于在 8 卡 MI300X 节点上通过 VLLM 加速 Llama 3.1 405B 的 FP8 推理。

来源:Hugging Face Blog · huggingface.co