PyTorch 博客· AMD: Xiaobing Zhang, Yutao Xu, Rui Li, Heng Yi, Xinya Zhang, Shijie Feng, Felix Li, Jeff Daily, Jiong Gong, Hattie Wu, Peng Sun, Emad Barsoum; META: Jason Ansel and Simon Layton·· 3 小时前AI 评分47
PyTorch 通过 FlyDSL 加速 AMD GPU 计算
Accelerating PyTorch on AMD GPUs with FlyDSL
AI 导读
PyTorch 用户可在 AMD MI350 系列 GPU 上通过 FlyDSL 使用 dense 和 grouped GEMM、fused multi-head attention 等操作，无需调用自定义内核。
来源：PyTorch 博客 · pytorch.org