跳到正文
GitHub · 灵魂体仓库·· 9 天前AI 评分45

amacharla15/fused-llm-inference-kernels-in-cuda:用 CUDA 实现 LLM 推理融合 kernel

amacharla15/fused-llm-inference-kernels-in-cuda

来源与原文

GitHub · 灵魂体仓库

发布时间:(北京时间)

阅读原始来源 · github.com ↗
完整链接与内容说明

https://github.com/amacharla15/fused-llm-inference-kernels-in-cuda

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

GitHub 上的 amacharla15/fused-llm-inference-kernels-in-cuda 项目提供一套用于 LLM 推理的高性能 CUDA kernel 实现,涵盖 warp/block reductions、激活函数,以及 fused RMSNorm、Softmax、RoPE 和 SwiGLU MLP block 等 20 个步骤。项目基于 Deep-ML 构建,可通过 python scaffold.py 运行,帮助学习让现代 Transformer 推理高效的 GPU 原语。

来源:GitHub · 灵魂体仓库 · github.com