gpu-kernels

标签

Cards List
#gpu-kernels

@no_stp_on_snek: 我的第二个且迟交的 Build Small 参赛作品。10天,1位开发者:从头构建的 Rust 引擎 + 自定义 GPU 内核 vs vLLM 在 N…

X AI KOLs Following ↗ · 2026-06-15 缓存

一位开发者从头构建了一个 Rust 推理引擎,带有自定义 GPU 内核,在 Nemotron-30B 解码上优于 vLLM,达到 75.7 vs 57 tok/s,提交至 Build Small 黑客马拉松。

0 人收藏 0 人点赞
#gpu-kernels

MiniMax Sparse Attention:百万令牌上下文(GitHub 仓库)

TLDR AI ↗ · 2026-06-15 缓存

MiniMaxAI 发布了 MSA,这是一个面向 NVIDIA SM100 GPU 优化的密集和稀疏注意力内核库,能够通过 FlashAttention 和稀疏 top-k 注意力高效处理百万令牌上下文。

0 人收藏 0 人点赞
#gpu-kernels

CODA: 将Transformer块重写为GEMM-尾声程序

Hacker News Top ↗ · 2026-05-22 缓存

介绍CODA,一种GPU内核抽象,将Transformer操作表达为GEMM加尾声程序以减少数据移动,覆盖Transformer块中几乎所有非注意力计算。

0 人收藏 0 人点赞
#gpu-kernels

@hamzaelshafie: 新深度博客文章:《剖析ThunderKittens:高性能AI内核的紧凑型DSL解剖》这篇帖子……

X AI KOLs Following ↗ · 2026-05-21 缓存

一篇详细分析ThunderKittens的博客文章,ThunderKittens是用于高性能AI内核的紧凑型DSL。文章包括从底向上的抽象分析,以及一个实现非因果注意力预填充内核的基准测试,该内核比FlashAttention-2快约1.55倍,与FlashAttention-3性能相当。

0 人收藏 0 人点赞
#gpu-kernels

Schanuel猜想与Triton的FPSan语义

Hacker News Top ↗ · 2026-05-16 缓存

FPSan是一个Triton编译器pass,通过用整数运算替换浮点运算来实现浮点程序代数等价性的验证,其正确性依赖于Schanuel猜想。

0 人收藏 0 人点赞
#gpu-kernels

@Modular: Mojo 拥有极简的样板代码、严格的类型系统以及编译时验证,所有这些都使其非常适合……

X AI KOLs Following ↗ · 2026-05-13 缓存

Modular 发布了开源的 Mojo 智能体技能,帮助 AI 编码智能体生成正确且地道的 Mojo 代码,包括一个将 CUDA 内核代码翻译为 Mojo 的演示。Mojo 的极简样板代码、严格的类型系统和编译时验证使其适用于智能体工作流。

0 人收藏 0 人点赞
#gpu-kernels

@leloykun:[进行中] 关于 Lean4-to-TileLang 张量程序超级优化器的博文:

X AI KOLs Following ↗ · 2026-05-12 缓存

一篇技术博文介绍了一种 Lean4-to-TileLang 张量程序超级优化器,能自动生成优化的 GPU/TPU 内核与超参数缩放规律,展示了相较 torch.compile 的性能提升。

0 人收藏 0 人点赞
#gpu-kernels

一个可定制的编译器,用于为AI模型生成高效的融合GPU内核 [P]

Reddit r/MachineLearning ↗ · 2026-05-11

作者介绍了一款用 Python 编写、高度可定制且易于修改的 ML 编译器。该编译器通过多级 IR 流水线将 LLMs 转换为优化的 CUDA 内核,在特定操作上实现了与 PyTorch 相当甚至更优的性能。文章详细阐述了该编译器的优化过程、降级规则以及用于生成高效融合 GPU 内核的 CLI 用法。

0 人收藏 0 人点赞
#gpu-kernels

KernelBench-X:评估LLM生成GPU内核的综合基准测试

Hugging Face Daily Papers ↗ · 2026-05-06 缓存

KernelBench-X是一个用于评估LLM生成GPU内核的新基准,揭示了任务结构对正确性的影响大于方法设计,且正确性并不保证硬件效率。

0 人收藏 0 人点赞
#gpu-kernels

AdaExplore:基于失败驱动的自适应与多样性保留搜索的高效内核生成

arXiv cs.CL ↗ · 2026-04-21 缓存

来自卡内基梅隆大学、华盛顿大学和Arm的研究人员提出了AdaExplore,这是一种用于GPU内核代码生成的LLM智能体框架。该框架通过失败驱动自适应与多样性保留搜索技术,在不进行额外微调的情况下,在KernelBench Level-2和Level-3基准测试中分别实现了3.12倍和1.72倍的加速。

0 人收藏 0 人点赞
#gpu-kernels

C++ CuTe / CUTLASS vs CuTeDSL (Python) in 2026 — 新的GPU内核/LLM推理工程师到底应该学什么?[D]

Reddit r/MachineLearning ↗ · 2026-04-20

讨论GPU内核工程从C++ CuTe/CUTLASS向NVIDIA基于Python的CuTeDSL的转变,质疑新工程师是应该学习遗留的C++模板,还是优先考虑为LLM推理工作而兴起的新技术栈。

0 人收藏 0 人点赞
#gpu-kernels

Block-sparse GPU kernels

OpenAI Blog ↗ · 2017-12-06 缓存

OpenAI 发布 block-sparse GPU kernels,这是一款用于在 GPU 上进行高效稀疏矩阵乘法的工具,可以减少神经网络操作的计算量和内存占用。

0 人收藏 0 人点赞
#gpu-kernels

tile-ai/tilelang

GitHub Trending (daily) ↗ · 2026-10-01 缓存

TileLang 是一个基于 TVM 构建的开源领域专用语言与编译器,用于编写高性能 GPU/CPU/NPU 内核,例如 GEMM 和 FlashAttention。近期新增了对 Huawei Ascend 950 的支持、LSP,以及面向 CUDA、ROCm 和 Metal 的多后端方言。

0 人收藏 0 人点赞
#gpu-kernels

deepseek-ai/DeepGEMM

GitHub Trending (daily) ↗ · 2026-04-21 缓存

# deepseek-ai/DeepGEMM 来源:[https://github.com/deepseek-ai/DeepGEMM](https://github.com/deepseek-ai/DeepGEMM) # DeepGEMM DeepGEMM 是一个统一的高性能张量核心内核库,整合了现代大语言模型的关键计算原语——GEMM(FP8、FP4、BF16)、带通信重叠的融合 MoE(Mega MoE)、闪电索引器的 MQA 评分、超连接(HC)等——形成一个统一的 CUDA 代码库。所有内核均在运行时编译

0 人收藏 0 人点赞
← Previous
← 返回首页

提交意见反馈