昇腾社区首页
中文
注册

函数:device_enable_peer_access

产品支持情况

产品

是否支持

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

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

x

Atlas 200/300/500 推理产品

x

功能说明

使能当前Device与指定Device之间的内存复制。使能内存复制是Device级的。

可提前调用acl.rt.device_can_access_peer接口查询当前Device与指定Device之间能否进行内存复制。需调用两次acl.rt.device_enable_peer_access接口使能两个Device之间的数据交互功能(例如,调用一次acl.rt.device_enable_peer_access接口使能Device 0到Device 1的数据交互,再调用一次acl.rt.device_enable_peer_access接口使能Device 1到Device 0的数据交互)。

使能Device间的数据交互功能后,若想关闭该功能,可调用acl.rt.device_disable_peer_access接口。

函数原型

  • C函数原型
    aclError aclrtDeviceEnablePeerAccess(int32_t peerDeviceId, uint32_t flags)
  • python函数
    1
    ret = acl.rt.device_enable_peer_access(peer_dev_id, flag)
    

参数说明

参数名

说明

peer_dev_id

int,Device ID,该ID不能与当前Device的ID相同。

用户调acl.rt.get_device_count接口获取可用的Device数量后,这个Device ID的取值范围:[0, (可用的Device数量-1)]。

flag

保留参数,当前必须设置为0。

返回值说明

返回值

说明

ret

int,错误码,返回0表示成功,返回其它值表示失败。