[控场AI]
· 3 分钟阅读· 1,958 字

PyTorch宣布成立Ray加速器特别兴趣小组

PyTorch宣布成立Ray加速器特别兴趣小组

PyTorch博客宣布成立Ray加速器SIG,推动异构计算支持规范化。

PyTorch官方博客宣布成立Ray在硬件加速器上的特别兴趣小组(SIG),标志着Ray对GPU、TPU等异构计算资源的支持从社区自发探索转向有组织推进。Ray诞生于十多年前的UC Berkeley,当时计算格局与如今显著不同,加速器已成为分布式负载的核心资源。SIG将协调设备发现、跨节点通信、内存管理及与PyTorch集成等分散问题,避免接口分裂。普通用户短期不会感知API变化,但中期有望获得更可预期的调度行为和性能特征。公告未披露治理结构和工作范围,开发者需关注后续社区更新。此举也反映AI基础设施中框架与硬件协同走向规范化的趋势。

rss source: Announcing the Special Interest Group (SIG) for Ray on Accelerators

PyTorch 官方博客近日发布公告,宣布成立针对 Ray 在硬件加速器上的特别兴趣小组(Special Interest Group,SIG)。这一动作标志着 Ray 项目在异构计算支持上从社区自发探索走向有组织的标准化推进。

从伯克利实验室到加速器时代

Ray 诞生于十多年前的 UC Berkeley,当时的设计假设与如今的计算环境已有明显差距。博客原文直言,Ray 创建时"计算格局看起来非常不同"。这种时代错位正是 SIG 成立的背景:当 GPU、TPU 等加速器成为分布式计算负载的核心资源,通用调度框架需要专门的兴趣小组来应对硬件碎片化带来的工程挑战。

SIG 的定位通常介于用户社区与核心维护者之间,负责收集需求、定义路线图、协调跨团队贡献。对于 Ray 这样被广泛用于机器学习训练和推理的框架,加速器支持的质量直接决定了上层应用的性能天花板。

Ray 的核心抽象是任务(task)和演员(actor),通过分布式调度器将 Python 函数和类实例分发到集群节点执行。其架构包含全局控制存储(GCS)、每个节点的 raylet 以及对象存储,最初面向 CPU 集群设计,资源模型以 CPU 核数为主。随着深度学习兴起,Ray 逐步引入自定义资源(custom resources)和加速器类型声明,但设备发现、拓扑感知调度和显存管理等能力仍依赖各集成方自行实现。这种历史包袱使得加速器支持呈现碎片化,正是 SIG 需要系统化梳理的起点。

为什么需要专门的 SIG

分布式框架与加速器的结合涉及多个层面的复杂问题:设备发现与资源建模、跨节点通信拓扑、内存管理策略、以及与 PyTorch 等上层框架的集成接口。这些问题分散在不同仓库和团队中,缺少统一协调容易导致接口分裂和重复造轮子。

SIG 模式在开源社区中有成熟先例,其价值在于把分散的讨论沉淀为可追踪的决议。PyTorch 生态本身也有多个 SIG,覆盖编译、分布式训练等方向,Ray on Accelerators 的加入补上了硬件资源管理这一环。

在开源治理中,SIG(特别兴趣小组)通常由社区成员自愿加入,围绕特定技术方向设立定期会议、邮件列表和决策记录。与 PMC(项目管理委员会)不同,SIG 不直接拥有代码仓库的最终否决权,而是向维护者提供建议和提案。PyTorch 生态中已有编译(torch.compile)、分布式训练等 SIG,其产出包括 RFC 文档、季度路线图和跨仓库的兼容性测试。Ray on Accelerators SIG 预计将沿用类似机制,把设备插件接口、多后端支持等议题纳入可追踪的协作流程。

对开发者的实际影响

普通用户短期内不会看到 API 层面的变化,但中期影响值得关注。SIG 的产出通常包括设计文档、兼容性矩阵和参考实现,这些会逐步反馈到 Ray 的发布节奏中。对于在生产环境部署 Ray 集群的团队,加速器相关的调度行为和性能特征有望变得更可预期。

说一下,公告本身只给出了 SIG 成立这一事实,具体的治理结构、参与方式和工作范围尚未在摘要中展开。关注该方向的开发者可以留意 PyTorch 博客和 Ray 社区仓库的后续更新,获取第一次会议纪要和工作组名单。

加速器资源建模在实践中涉及几个具体接口:Ray 通过 num_gpus 等逻辑资源声明来调度 GPU,但无法感知 NVLink 拓扑或 MIG 分区;设备插件机制则允许第三方注册自定义加速器类型。跨节点通信方面,Ray 的对象存储传输默认走 TCP,对 GPU 直连(GPUDirect RDMA)的支持仍在演进。内存管理上,Ray 的对象溢出(object spilling)主要面向主机内存,显存回收依赖上层框架。SIG 若能在这些接口上形成规范,将直接影响多加速器后端的实现一致性。

生态信号:加速器支持走向规范化

从更大的视角看,这一 SIG 的成立反映了 AI 基础设施领域的一个趋势:框架与硬件的协同不再是边缘问题,而是决定平台竞争力的核心变量。PyTorch 选择在官方博客发布该消息,也说明 Ray 在 PyTorch 生态中的位置被进一步确认。

对于关注分布式训练和推理部署的工程师,建议持续跟踪 SIG 的输出,尤其是设备插件规范和多加速器后端支持方面的进展。这些内容往往比版本号更新更能反映项目的真实方向。

#Ray#PyTorch#SIG#硬件加速器#分布式计算#开源框架
分享:

相关推荐