一言
发表于
一言发生错误
ForgeKernel with PyPTO
发表于
前两天看到了 OpenBMB 开源的 ForgeKernel,一个让 agent 自动优化算子的工作,在 FlashInfer-Bench(B200)、CANN-Bench (Ascend A3 & 950DT) 上都取得了第一名。这个进步还是很夸张的。笔者之前也参与过算子的竞赛活动,尝试过用 AI 自己迭代写算子,但是没能取得很好的成绩,看到这样的工作自然忍不住想学习一下。正好最近在看 PyPTO,就想能不能复现这个工作,使用 PyPTO 的后端。
需要说明的是,官方仓库目前还没有放出开源的 Ascend Agent 框架,本文中提到的实现都是笔者根据论文的描述以及已开源的 NVIDIA 实现模仿实现的,代码质量也是 vibe 质量,欢迎批评指正。
Decode长度预测的误差下界
发表于
生成armv4汇编
发表于
显然学习体系结构能随时生成汇编非常方便。教学使用的体系结构是 armv4,不过现代编译器默认生成的已经是 armv8 了,连指令宽度都不一样。有必要学习怎么编译和查看 armv4 编译。
MP伪逆和多元线性回归
发表于 分类于 数学