PPyTorch Blog
PyTorch 基金会发布 PyTorch 2.15,所有使用 nccl 的进程组默认改用重写后的 NCCL 后端,并附带在集合通信挂起时转储现场信息的 flight recorder,可通过 TORCH_DIST_USE_NCCL2=0 或 nccl-legacy 回退。该版本还为 Apple Silicon 加入 float8_e4m3fn 与 _scaled_mm 支持,新的 grouped_mm 在 M3 Ultra 上让 MoE 解码吞吐提升 1.32–1.34 倍。默认 wheel 升级到 CUDA 13.2,同时提供 CUDA 13.4 wheel 并实验性支持 NVIDIA Rubin GPU,另新增 ROCm 10.0 wheel,且不再支持 Python 3.10 和 macOS 14。
PyTorch 基金会发布 PyTorch 2.15,所有使用 nccl 的进程组默认改用重写后的 NCCL 后端,并附带在集合通信挂起时转储现场信息的 flight recorder,可通过 TORCH DIST USE NCCL2=0 或 nccl-legacy 回退。该版本还为 Apple Silicon 加入 float8 e4m3fn 与 scaled mm 支持,新的 grouped mm 在 M3 Ultra 上让 MoE 解码吞吐提升 1.32–1.34 倍。默认 wheel 升级到 CUDA 13.2,同时提供 CUDA 13.4 wheel 并实验性支持 NVIDIA Rubin GPU,另新增 ROCm 10.0 wheel,且不再支持 Python 3.10 和 macOS 14。
匹配字段摘要、正文