昇腾社区首页
中文
注册

函数:get_device

产品支持情况

产品

是否支持

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

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

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

获取当前正在使用的Device的ID。

函数原型

  • C函数原型
    1
    aclError aclrtGetDevice(int32_t *deviceId)
    
  • python函数
    1
    device_id, ret = acl.rt.get_device()
    

参数说明

参数名

说明

device_id

Device ID的指针。

返回值说明

返回值

说明

ret

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

约束说明

如果没有提前指定Device,则调用aclrtGetDevice接口时,返回错误。指定Device的方式包括:在init接口中开启默认Device、调用acl.rt.set_device接口显式指定Device、调用acl.rt.create_context接口隐式指定Device。