函数：set_device

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

√

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

√

Atlas 训练系列产品

√

Atlas 推理系列产品

√

Atlas 200I/500 A2 推理产品

√

Atlas 200/300/500 推理产品

√

aclError aclrtSetDevice(int32_t deviceId)

ret = acl.rt.set_device(device_id)

指定当前线程中用于运算的Device，同时隐式创建默认Context。

对于 Atlas 200/300/500 推理产品 ，该默认Context中包含2个Stream，1个默认Stream和1个执行内部同步的Stream。

对于 Atlas 200I/500 A2 推理产品 ，该默认Context中包含1个默认Stream。

对于 Atlas 训练系列产品 ，该默认Context中包含1个默认Stream。

对于 Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件 ，该默认Context中包含1个默认Stream。

对于 Atlas 推理系列产品 ，在标准形态下，该Context中包含2个Stream，1个默认Stream和1个执行内部同步的Stream。

对于 Atlas A3 训练系列产品/Atlas A3 推理系列产品 ，调用本接口会隐式创建默认Context，该默认Context中包含1个默认Stream。

device_id：int，Device设备号。

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

ret：int，错误码。

• 返回0表示成功。
• 返回其它值表示失败。

如果多次调用acl.rt.set_device接口而不调用acl.rt.reset_device接口释放本进程使用的Device资源，功能上不会有问题，因为在进程退出时也会释放本进程使用的Device资源。建议acl.rt.set_device接口和acl.rt.reset_device接口配对使用，在不使用Device上资源时，通过调用acl.rt.reset_device接口及时释放本进程使用的Device资源。

在不同进程或线程中支持调用acl.rt.set_device接口指定同一个Device用于运算。在同一个进程中的多个线程中，如果调用acl.rt.set_device接口指定同一个Device用于运算，这时隐式创建的默认Context是同一个。

接口调用流程与示例，请参见运行时资源申请与释放、同步等待。