跳转至

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 分布式编排的基础。