patch_for_hcom

针对PyTorch 2.1版本中不支持入图的集合通信算子提供的补丁函数，实现部分集合通信算子入图。

def patch_for_hcom()

无

无

无

      
           # 1.导入patch_for_hcom包
import torch_npu
import torchair
from torchair import patch_for_hcom
# 2.在图执行之前调用patch方法
patch_for_hcom()

父主题： 基本类和接口