patch_for_hcom

功能说明

针对PyTorch 2.1版本，补齐PyTorch原生部分集合通信算子无法入图功能。

函数原型

def patch_for_hcom()

参数说明

无

返回值说明

无

约束说明

无

调用示例

1
2
3
4

import torch_npu
import torchair
# 在图执行之前调用patch方法
torchair.patch_for_hcom()