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