get_world_rank_from_group_rank

Description

Obtains the world rank ID based on the rank ID of the process in the group.

Prototype

def get_world_rank_from_group_rank(group, group_rank_id)

Parameters

Parameter

Input/Output

Description

group

Input

A string containing a maximum of 128 bytes, including the end character.

Group name, which can be a user-defined value or hccl_world_group.

group_rank_id

Input

An int.

Rank ID of a process in a group.

Returns

Rank ID of the process in the global group (hccl_world_group), which is of the int type.

Constraints

  • This API must be called after the initialization of collective communication is complete.
  • The caller rank must be within the range defined by the group argument passed to this API call. Otherwise, the API call fails.
  • After the create_group call, call this API to convert the group rank ID to the world rank ID.

Applicability

Atlas Training Series Product

Example

The following is only a code snippet and cannot be executed. For details about how to call the HCCL Python APIs to perform collective communication, see Sample Code.

1
2
3
from npu_bridge.npu_init import *
create_group("myGroup", 4, [0, 1, 2, 3])
worldRankId = get_world_rank_from_group_rank("myGroup", 1)