跳到正文
原文
AGI Hunt· PyTorch·· 2 小时前AI 评分25

AMD 将在 PyTorch 大会展示 FlyDSL:基于 MLIR 的 GPU 内核新后端

AMD 将在 PyTorch 大会展示 FlyDSL:基于 MLIR 的 GPU 内核新后端

AI 导读

AMD 将在 PyTorch 大会上介绍 Python 原生、基于 MLIR 的 GPU 内核 DSL FlyDSL。该 DSL 将集成进 TorchInductor 的编译与自动调优管线,同时保留 torch.compile 的用户体验。演讲还将对比 Triton 与 FlyDSL 在 AMD Instinct GPU 上 transformer 训练与推理的性能。

来源:AGI Hunt · agihunt.info