跳到正文
Hugging Face Blog·· 2026-06-11AI 评分55

Hugging Face 博客:PyTorch Profiling 系列第二篇,从 nn.Linear 到融合 MLP

Profiling in PyTorch (Part 2): From nn.Linear to a Fused MLP

AI 导读

Hugging Face Blog 发布 PyTorch Profiling 系列第二篇,用 profiler trace 对比 nn.Linear 与三层 GeGLU MLP 在 eager、torch.compile 和 Liger 手写 kernel 三种设置下的表现。

来源:Hugging Face Blog · huggingface.co