HcclCommSymWinGet

产品支持情况

功能说明

根据已注册对称内存的虚拟地址指针，返回对应的窗口资源句柄及其在窗口内的偏移量。

函数原型

1

HcclResult HcclCommSymWinGet(HcclComm comm, void *ptr, size_t size, CommSymWindow *winHandle, size_t *offset)

参数说明

返回值

HcclResult：接口成功返回HCCL_SUCCESS，其他失败。

约束说明

• 仅支持Atlas A3 训练系列产品/Atlas A3 推理系列产品的超节点内通信。
• 仅支持通信算子展开模式为AI CPU的场景。

调用示例

// 创建并初始化通信域配置项
HcclCommConfig config;
HcclCommConfigInit(&config);
// 按需修改通信域配置
config.hcclSymWinMaxMemSizePerRank = 10; //单位GB, 默认值为16。设置对称堆预留的虚拟内存大小 = rankSize * config.hcclSymWinMaxMemSizePerRank;
// 初始化集合通信域
HcclComm hcclComm;
HCCLCHECK(HcclCommInitRootInfoConfig(rankSize, &rootInfo, deviceId, &config, &hcclComm));

// 物理内存属性配置
int32_t deviceId;
aclrtGetDevice(&deviceId);
aclrtPhysicalMemProp prop;
prop.handleType = ACL_MEM_HANDLE_TYPE_NONE;
prop.allocationType = ACL_MEM_ALLOCATION_TYPE_PINNED;
prop.memAttr = ACL_HBM_MEM_HUGE;
prop.location.id = deviceId;
prop.location.type = ACL_MEM_LOCATION_TYPE_DEVICE;
prop.reserve = 0;
// 获取对齐粒度，通常为2M
size_t granularity = 0;
aclrtMemGetAllocationGranularity(&prop, ACL_RT_MEM_ALLOC_GRANULARITY_RECOMMENDED, &granularity);
// size按粒度对齐
size_t size = 2 * 1024 * 1024;
size_t allocSize = (size + granularity - 1) / granularity * granularity;
// 预留虚拟内存
void *virPtr = nullptr;
aclrtReserveMemAddress(&virPtr, allocSize, 0, nullptr, 1);
// 申请物理内存
aclrtDrvMemHandle handle;
aclrtMallocPhysical(&handle, allocSize, &prop, 0);
// 建立物理到虚拟的映射
aclrtMapMem(virPtr, allocSize, 0, handle, 0);

CommSymWindow sym_win;
// 注册对称内存
HcclCommSymWinRegister(hcclComm, virPtr, allocSize, &sym_win, HCCL_WIN_COLL_SYMMETRIC);
// 使用HcclCommSymWinGet获取对称内存资源
CommSymWindow temp_win;
size_t offset = 0;
HcclCommSymWinGet(hcclComm, virPtr + 10, 1024, &temp_win, &offset);
// 解注册对称内存
HcclCommSymWinDeregister(sym_win);
// 释放内存
aclrtUnmapMem(virPtr);
aclrtFreePhysical(handle);
aclrtReleaseMemAddress(virPtr);
// 销毁通信域
HcclCommDestroy(hcclComm);