编译图 API#

输入和输出节点#

ray.dag.input_node.InputNode

在 DAG 构建 API 中用于标记 DAG 入口点的 Ray DAG 节点。

ray.dag.output_node.MultiOutputNode

在 DAG 构建 API 中用于标记 DAG 出口点的 Ray DAG 节点

DAG 构建#

ray.actor.ActorMethod.bind

为 Ray DAG 构建将参数绑定到 actor 方法。

ray.dag.DAGNode.with_tensor_transport

为此节点配置 torch 张量传输。

ray.experimental.compiled_dag_ref.CompiledDAGRef

对编译后的 DAG 执行结果的引用。

编译图操作#

ray.dag.DAGNode.experimental_compile

为此 DAG 编译一个加速执行路径。

ray.dag.compiled_dag_node.CompiledDAG.execute

使用编译后的执行路径执行此 DAG。

ray.dag.compiled_dag_node.CompiledDAG.visualize

通过显示任务及其依赖关系来可视化编译图。

配置#

ray.dag.context.DAGContext

Ray DAG 的全局设置。