调用梯度切分类接口设置反向计算阶段的allreduce切分融合策略。
set_split_strategy_by_idx:基于梯度的索引id,在集合通信group内设置反向梯度切分策略。
from npu_bridge.npu_init import * set_split_strategy_by_idx([20, 100, 159])
set_split_strategy_by_size:基于梯度数据量百分比,在集合通信group内设置反向梯度切分策略。
from npu_bridge.npu_init import * set_split_strategy_by_size([60, 20, 20])