yr.FunctionGroupContext

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

Bases: object

A context class for managing function group information.

Attributes

device_name	Name of the device used by this function instance, e.g., NPU/Ascend910B.
rank_id	The ID of this function instance within the function group.
world_size	Total number of function instances in the group.
server_list	Server info list for inter-instance communication.

Methods

__init__