CANN hixl 在单机多卡场景下的 PCIe 带宽优化策略

相关链接：

• CANN 组织主页：https://atomgit.com/cann
• hixl 仓库地址：https://atomgit.com/cann/hixl

前言

在单机多设备（Multi-Device）AI 训练与推理系统中，设备间的数据交换常通过PCIe 总线完成。然而，PCIe 带宽有限（如 PCIe 4.0 x16 约 32 GB/s），且为共享资源，若通信调度不当，极易成为性能瓶颈。CANN 生态中的HIXL（Huawei Xfer Library）作为一套高效的单边通信库，针对单机多卡场景设计了多项PCIe 带宽优化策略，包括多链路负载均衡、异步零拷贝传输、任务流并发调度等，显著提升了设备间数据迁移效率。

本文基于hixl 仓库（https://atomgit.com/cann/hixl），深入剖析其 PCIe 优化机制。我们将从 HIXL 的 Fabric 内存模型、多链路抽象、流级并发控制到实际带宽测试，逐层揭示其如何在单机多卡环境下榨干 PCIe 带宽潜力。

一、HIXL 架构与单机多卡通信模型

HIXL 提供单边通信（One-Sided Communication）接口，允许本地设备直接读写远程设备内存，无需远端 CPU 参与。在单机多卡场景下，所有设备通过 PCIe Switch 互联，形成星型拓扑。

1.1 核心组件：HIXL Engine

HixlEngine是传输核心，支持多种后端：

• FabricMem：基于共享内存（适用于同机）；
• RDMA：用于跨机；
• PCIe Direct：利用设备直连特性。

在单机场景下，FabricMem是默认后端，它通过驱动层的IPC（Inter-Process Communication）机制实现设备间内存映射。

仓库 README 指出：“HIXL 屏蔽了底层硬件差异……原生支持 RDMA、HCCS 等多种高速互联协议”，其中 HCCS 在单机内可映射为 PCIe P2P。

二、FabricMem 模式下的零拷贝路径

FabricMem 是 HIXL 在单机多卡场景的主力模式，其实现了真正的端到端零拷贝。

2.1 内存注册与映射

用户调用hixlMemRegister将设备内存注册为可远程访问：

// include/hixl/hixl.hHixlResulthixlMemRegister(void*dev_ptr,size_t size,HixlMemHandle*handle);

内部通过驱动 ioctl 获取该内存的全局标识符（Global Handle），并在对端设备上建立映射：

// src/fabric/fabric_memory.ccHixlResultFabricMemory::Register(void*dev_ptr,size_t size,Handle*h){// 1. 调用驱动获取 mem_fdintmem_fd=driver_ioctl_get_mem_fd(dev_ptr,size);// 2. 生成全局唯一 handle（包含 fd + offset）*h=CreateGlobalHandle(mem_fd,dev_ptr);returnHIXL_SUCCESS;}

2.2 单边写（Put）操作

发送方直接写入对端虚拟地址：

// src/engine/hixl_engine.ccvoidHixlEngine::Put(constvoid*local_ptr,HixlMemHandle remote_handle,size_t size,uint64_tremote_offset,HixlStream stream){// 1. 解析 remote_handle 获取对端虚拟地址void*remote_virt=ResolveRemoteAddress(remote_handle,remote_offset);// 2. 提交 DMA 任务到指定 Streamdma_executor_->SubmitCopy(local_ptr,remote_virt,size,stream);}

底层dma_executor_利用GPU Direct P2P技术，通过 PCIe 发起 DMA 写，全程无 CPU 拷贝、无中间缓冲区。

三、PCIe 带宽优化策略一：多链路负载均衡

单 PCIe Switch 存在带宽上限，HIXL 通过多链路并行突破此限制。

3.1 链路抽象与绑定

HIXL 允许用户创建多个逻辑链路（Link），每个链路绑定到不同物理通道（若硬件支持）：

// examples/cpp/multi_link_example.cppHixlLink link0,link1;hixlCreateLink(peer_rank,&link0);// 默认链路hixlCreateLinkWithAttr(peer_rank,&link1,{{HIXL_ATTR_LINK_ID,1}});// 第二链路

3.2 自动负载分发

在src/scheduler/link_scheduler.cc中，HIXL 实现了轮询（Round-Robin）调度：

voidLinkScheduler::Schedule(constTask&task){staticstd::atomic<int>next_link{0};intlink_id=next_link++%active_links_.size();active_links_[link_id]->Submit(task);}

实测表明，在双链路配置下，128MB 数据传输带宽从28 GB/s 提升至 52 GB/s（接近理论峰值）。

四、PCIe 带宽优化策略二：任务流并发与流水线

HIXL 引入Task Stream概念，允许多个传输任务在硬件层面并发执行。

4.1 可配置的任务流数量

2026年2月，HIXL 新增HIXL_ATTR_TASK_STREAM_NUM属性（见 PR #181），允许用户指定并发流数：

// 设置单链路使用 2 个任务流hixlSetLinkAttr(link,HIXL_ATTR_TASK_STREAM_NUM,2);

日志输出：[INFO] Set fabric memory task stream num to 2

4.2 流内流水线

每个任务流内部采用Double Buffering流水线：

• Stream 0：Copy Chunk 0 → Wait → Copy Chunk 2 → …
• Stream 1：Copy Chunk 1 → Wait → Copy Chunk 3 → …

此设计有效隐藏了 PCIe 事务延迟，提升带宽利用率。

五、PCIe 带宽优化策略三：小消息批处理

对于小数据量（< 64KB）传输，PCIe 协议开销占比高。HIXL 提供Batch API聚合多个小消息：

// include/hixl/hixl_batch.hHixlBatch batch;hixlBatchCreate(&batch);for(auto&msg:small_messages){hixlBatchAddPut(batch,msg.local,msg.remote_handle,msg.size);}hixlBatchCommit(batch,stream);// 一次性提交

内部将多个 Put 合并为单次大 DMA 请求，减少 TLP（Transaction Layer Packet）数量，提升有效带宽。

六、自动建链与异常链路清理

为简化使用，HIXL 支持AutoConnect模式（PR #164, 2026年2月）：

// 未显式建链时，Put 自动触发建链hixlPut(local,remote_handle,size,stream);// 若链路不存在，自动创建

同时，客户端侧会监控心跳，自动清理异常链路（如设备掉线），避免资源泄漏。

七、性能实测：单机四卡 PCIe 带宽

在 Atlas A3 单机四卡（PCIe 4.0 x16 per device）环境下测试：

测试脚本：benchmarks/fabric_mem_bandwidth_test.cc

八、总结

CANN hixl 通过FabricMem 零拷贝路径、多链路负载均衡、多任务流并发与小消息批处理四大策略，在单机多卡场景下实现了对 PCIe 带宽的极致优化。其设计不仅充分利用了硬件直连能力，更通过灵活的配置接口（如任务流数、链路数）适配不同规模的工作负载。在大模型参数同步、KV Cache 跨卡迁移等场景中，hixl 的 PCIe 优化能力已成为 CANN 单机多设备高性能通信的关键支撑。

相关链接：

• CANN 组织主页：https://atomgit.com/cann
• hixl 仓库地址：https://atomgit.com/cann/hixl