arXiv · cs.AI · Luk Koska, Tom Koska, Ben Koska · 5 小时前模型评分45
用 Lean 形式化 PTX 以证明 图形处理器（GPU，Graphics Processing Unit） 内核正确性
Formalizing PTX in Lean to Prove GPU Kernel Correctness
导读
研究者提出在 Lean 中形式化 PTX 的方法，可对深度学习 图形处理器（GPU，Graphics Processing Unit） 内核中常见的 acquire-release、tensor cores、代理、异步操作、多内核流水线和数据相关控制流建模与推理。他们用该方法证明了 WGMMA GEMM 内核、FlashAttention 和 3 内核 MoE 流水线简单实现的正确性。
分类置信度
中 · 待人工复核
- 存在未解决的相邻类别：compute-hardware，需要人工复核。
- 只有单一媒体来源，需要人工复核。
来源级别：媒体与个人
事件类型：研究发布
证据摘录
- “We present an approach to mathematically formalizing PTX, an approach that allows modeling and reasoning about patterns common in GPU kernels for Deep Learning”
来源：arXiv · cs.AI · arxiv.org