Specifying Devices to Perform Collective Communication
Call create_group to specify the devices that participate in collective communication. If this API is not called to create a custom group, all devices are created as the global hccl_world_group by default based on the ranktable file.
For example, you can select the first two devices in hccl_world_group as a group for collective communication:
from hccl.manage.api import create_group create_group("myGroup", 2, [0, 1])