开发者
资源

HcommMemReg

产品支持情况

产品

是否支持

Atlas 350 加速卡

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

功能说明

注册内存到指定EndPoint。

函数原型

1
HcclResult HcommMemReg(EndpointHandle endpointHandle, const char *memTag, HcommMem mem, void **memHandle)

参数说明

参数名

输入/输出

说明

endpointHandle

输入

Endpoint句柄。

EndpointHandle类型的定义请参见EndpointHandle

memTag

输入

内存字符串标识。

mem

输入

内存描述信息,包含内存物理位置类型、内存地址、内存区域字节数。

HcommMem类型的定义请参见HcommMem

memHandle

输出

注册内存句柄

返回值

HcclResult:接口成功返回HCCL_SUCCESS,其他失败。

约束说明

调用示例

const EndpointDesc endpointDesc = {
    .protocol = COMM_PROTOCOL_ROCE,
    .commAddr = {
        .type = COMM_ADDR_TYPE_IP_V4,
        .addr = {{192, 168, 1, 100}}
    },
    .loc = {
        .locType = ENDPOINT_LOC_TYPE_DEVICE,
        .device = {
            .devPhyId = 0,
            .superDevId = 0,
            .serverIdx = 0,
            .superPodIdx = 0
        }
    },
    .raws = {0}
};
EndpointHandle endpointHandle = nullptr;
HcclResult result = HcommEndpointCreate(&endpointDesc, &endpointHandle);
const char *memTag = "HcclBuffer";
HcommMem mem = {
    .type = HCCL_MEM_TYPE_DEVICE,
    .addr = reinterpret_cast<void*>(0x1111),
    .size = 100
};
void *memHandle;
result = HcommMemReg(endpointHandle, memTag, mem, &memHandle);