跳到主要内容
检索

# FlyDSL GPU内核DSL

2 篇关联报道

订阅更新
  1. PyTorch

    PyTorch 2.15把重写的NCCL后端设为默认:集合通信卡住时,每个rank自动落盘现场

    PyTorch 2.15把2.14中作为可选nccl2预览的重写NCCL后端设为所有nccl进程组的默认,并保留TORCH_DIST_USE_NCCL2=0与nccl-legacy两条回退路径。此前集群规模下单个GPU停滞就可能拖停整个训练,定位原因往往要靠手工复现;新版本加入与后端无关的飞行记录器,集合通信挂起时向磁盘写出事后现场,任一rank超时都会触发所有rank转储。

  2. PyTorch

    AMD GPU 上的 PyTorch 多了一条后端:FlyDSL 让算子按硬件显式排布,vLLM 整请求最高 3.01 倍

    AMD MI350 系列 GPU 上的 PyTorch 用户现在可以通过既有算子接口调用 FlyDSL 内核,覆盖稠密与分组 GEMM、MXFP8/MXFP4 缩放 GEMM、RMSNorm、TopK 以及经 AOTriton 的注意力。此前这些算子只能走 ATen 或 Triton 等实现;在 MI355X 的算子级测试中,FlyDSL 对 15 个 BF16 稠密 GEMM 形状相对逐形状更快的 ATen/Triton 取得 1.10 倍几何平均加速,MXFP8 与 MXFP4 相对 ATen 分别为 1.58 倍和 1.68 倍。端到端 vLLM 测试中整请求加速最高为 BF16 的 1.13 倍、MXFP8 的 2.04 倍和 MXFP4 的 3.01 倍,具体取决于模型与并发度。

已读完全部报道