每天听简报,了解最新科技、AI、软件资讯
AMD的开源项目FlyDSL,让开发者用Python编写显卡内核,明确描述数据怎样分块、搬运,以及线程怎样分工。它基于MLIR编译栈,把这些描述转成AMD显卡执行的代码。项目提供向量加法和矩阵乘法示例,既可安装Python包,也能从源码构建。使用环境是Linux,执行显卡测试需要ROCm,Python版本至少三点一零。PyTorch十月二日预告,会议将介绍它与TorchInductor编译、自动调优流程的集成,保持torch.compile的用法;还计划对比Triton实现,但预告没有性能数值。官方文档跟随主分支更新,接口可能领先已发布安装包,使用时应核对版本。
进度保存在本设备 · 登录同步