yr.FunctionGroupContext

yr.FunctionGroupContext#

class yr.FunctionGroupContext(rank_id: int = 0, world_size: int = 0, server_list: List[yr.config.ServerInfo] = <factory>, device_name: str = '')#

基类:object

用于管理函数组信息的上下文。

属性

device_name

挂载到此函数实例的设备的名称,例如,NPU/Ascend910B。

rank_id

本函数实例在函数组中的 id。

world_size

函数组中函数实例数。

server_list

函数组中每个函数实例集合通信的 server 信息列表。

方法