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