arXiv · AI· Luk Koska, Tom Koska, Ben Koska·· 3 小时前AI 评分28
在 Lean 中形式化 PTX 以证明 GPU 内核正确性
Formalizing PTX in Lean to Prove GPU Kernel Correctness
AI 导读
研究人员提出在 Lean 中数学形式化 PTX 的方法，用于建模和推理 GPU 内核中的常见模式。 该方法已用于证明 WGMMA GEMM 内核、FlashAttention 及 3 内核 MoE 流水线的简单实现正确性。
来源：arXiv · AI · arxiv.org