ray_utils.py — Ray 分布式计算工具¶
文件路径: verl/utils/ray_utils.py
文件概述¶
封装 Ray 分布式计算框架的工具函数。verl 使用 Ray 进行多节点编排:启动 Actor worker、Critic worker、Rollout worker 等,并在它们之间调度任务。
核心功能¶
- Ray Actor 创建和管理
- 资源分配(GPU/NPU 数量)
- Worker 间通信辅助
- Ray cluster 状态检查
与其他模块的关系¶
- 依赖
device.py的get_resource_name()确定资源类型 - 被 trainer 模块用来创建和管理分布式 worker
小结¶
Ray 框架的封装工具,是 verl 分布式编排的基础。