针对PyTorch 2.1版本中不支持入图的集合通信算子提供的补丁函数,实现部分集合通信算子入图。
def patch_for_hcom()
无
1 2 3 4 5 6
# 1.导入patch_for_hcom包 import torch_npu import torchair from torchair import patch_for_hcom # 2.在图执行之前调用patch方法 patch_for_hcom()