指定要进行集合通信的设备

调用create_group来指定要进行集合通信的设备。如果不调用此接口创建用户自定义group,则按ranktable文件默认将所有设备创建为全局的hccl_world_group。

例如选择hccl_world_group中的前两个npu昇腾AI处理器作为group进行集合通信:

from npu_bridge.npu_init import *  
create_group(“myGroup”, 2, [0, 1])