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

Hugging Face 发布针对 AMD MI300X 的自定义内核优化实践

Creating custom kernels for the AMD MI300

AI 导读

Hugging Face 与 AMD 合作开源了针对 MI300X 的自定义内核,通过融合 RMS norm 与 FP8 转换、优化 SwiGLU 计算以及针对低批次大小的 Skinny GEMM 拆分策略,显著提升了 Llama 3.1 405B 在 VLLM 中的推理性能。

推荐理由

文章详细拆解了针对MI300X的三种自定义内核优化技巧,提供了可复用的工程实践方法。

来源:Hugging Face Blog · huggingface.co