TransferAsync

产品支持情况

说明：针对 Atlas A2 训练系列产品 / Atlas A2 推理系列产品 ，仅支持Atlas 800I A2 推理服务器、Atlas 300I A2 推理卡、A200I A2 Box 异构组件。

函数功能

与远端HIXL进行批量异步内存传输。

函数原型

1
2
3
4
5

Status TransferAsync(const AscendString &remote_engine,
                     TransferOp operation,
                     const std::vector<TransferOpDesc> &op_descs,
                     const TransferArgs &optional_args,
                     TransferReq &req)

参数说明

调用示例

//初始化客户端和服务端engine，并完成链接
client_engine.TransferAsync(remote_engine, operation, op_descs, optional_args, req);

返回值

• SUCCESS：成功
• NOT_CONNECTED：没有与对端创建链接
• RESOURCE_EXHAUSTED：资源耗尽
• 其他：失败

约束说明

• 调用该接口之前，需要先调用Connect接口完成与对端的建链。
• 该接口需要和Initialize运行在同一个线程上，如需切换线程调用该接口，需要在Initialize所在线程调用“aclrtGetCurrentContext”获取context，并在新线程调用“aclrtSetCurrentContext”设置context。
• 当前异步传输只支持直传，暂不支持中转传输，默认直传。