HCCL 代码示例

本目录提供了不同场景下使用 HCCL 接口实现集合通信功能的示例代码。

点对点通信

• HcclSend/HcclRecv（基础收发功能）
• HcclBatchSendRecv（实现 Ring 环状通信）

集合通信

• AllReduce
• Broadcast
• AllGather
• ReduceScatter
• Reduce
• AlltoAll
• AlltoAllV
• AlltoAllVC
• Scatter

AI 框架

• PyTorch
• Tensorflow

自定义点对点通信算子

• 自定义 Send/Recv 算子（基于 AICPU 通信引擎）