函数:get_run_mode
函数:reset_device
| C函数原型 | aclError aclrtResetDevice(int32_t deviceId) |
|---|---|
| Python函数 | ret = acl.rt.reset_device(device_id) |
| 函数功能 | 复位当前运算的Device,释放Device上的资源,包括默认Context、默认Stream以及默认Context下创建的所有Stream,同步接口。若默认Context或默认Stream下的任务还未完成,系统会等待任务完成后再释放。 |
| 输入说明 | device_id:int,Device设备号。 |
| 返回值说明 | ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 若要复位的Device上存在显式创建的Context、Stream、Event,在复位前,建议遵循如下接口调用顺序,否则可能会导致业务异常。 接口调用顺序:调用acl.rt.destroy_event接口释放Event/调用acl.rt.destroy_stream接口释放显式创建的Stream-->调用acl.rt.destroy_context释放显式创建的Context-->调用acl.rt.reset_device接口。 |
| 参考资源 | 接口调用流程与示例,请参见运行管理资源申请与释放、同步等待。 |
父主题: Device管理
函数:get_device
| C函数原型 | aclError aclrtGetDevice(int32_t *deviceId) |
|---|---|
| Python函数 | device_id, ret = acl.rt.get_device() |
| 函数功能 | 获取当前正在使用的Device的ID,同步接口。 |
| 输入说明 | 无 |
| 返回值说明 | device_id:int,正在使用的Device设备号。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 如果没有调用acl.rt.set_device接口显式指定Device或没有调用acl.rt.create_context接口隐式指定Device,则调用acl.rt.get_device接口时,返回错误。 |
父主题: Device管理
函数:get_run_mode
| C函数原型 | aclError aclrtGetRunMode(aclrtRunMode *runMode) |
|---|---|
| Python函数 | run_mode, ret = acl.rt.get_run_mode() |
| 函数功能 | 获取当前昇腾AI软件栈的运行模式,同步接口。 |
| 输入说明 | 无 |
| 返回值说明 | run_mode:int,表示运行模式。 - 0:ACL_DEVICE,昇腾AI软件栈运行在Device的Control CPU或板端环境上。 - 1:ACL_HOST,昇腾AI软件栈运行在Host CPU上。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
| 参考资源 | 接口调用流程与示例,请参见运行管理资源申请与释放。 |
父主题: Device管理
函数:set_ts_device
| C函数原型 | aclError aclrtSetTsDevice(aclrtTsId tsId) |
|---|---|
| Python函数 | ret = acl.rt.set_ts_device(ts_id) |
| 函数功能 | 设置本次计算需要使用的Task Schedule,仅昇腾AI处理器支持vector core计算单元才生效,同步接口。 |
| 输入说明 | ts_id:int,指定本次计算需要使用的Task Schedule。如果昇腾AI软件栈中仅有AICORE Task Schedule,则设置该参数无效,默认使用AICORE Task Schedule。 - 0:ACL_TS_ID_AICORE,使用AICORE Task Schedule。 - 1:ACL_TS_ID_AIVECTOR,使用AIVECTOR Task Schedule。 - 2:ACL_TS_ID_RESERVED。 |
| 返回值说明 | ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
父主题: Device管理
函数:get_device_count
| C函数原型 | aclError aclrtGetDeviceCount(uint32_t *count) |
|---|---|
| Python函数 | count, ret = acl.rt.get_device_count() |
| 函数功能 | 获取可用Device的数量,同步接口。 |
| 输入说明 | 无 |
| 输出说明 | 无 |
| 返回值说明 | count:int,获取Device数量。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
父主题: Device管理
函数:get_device_utilization_rate
| C函数原型 | aclError aclrtGetDeviceUtilizationRate(int32_t deviceId, aclrtUtilizationInfo *utilizationInfo) |
|---|---|
| Python函数 | utilization_info, ret = acl.rt.get_device_utilization_rate (device_id) |
| 函数功能 | 查询Device上Cube、Vector、AI CPU等的利用率。同步接口。 |
| 输入说明 | device_id:int,Device ID。 用户调用acl.rt.get_device_count接口获取可用的Device数量后,这个Device ID的取值范围:[0, (可用的Device数量-1)]。 |
| 输出说明 | 无 |
| 返回值说明 | utilization_info:dict,aclrtUtilizationInfo类型利用率信息字典,具体请参见aclrtUtilizationInfo。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | - 昇腾虚拟化实例场景下,不支持调用本接口查询利用率,接口返回值无实际意义。 - 开启Profiling功能时,不支持调用本接口查询利用率,接口返回值无实际意义。 - 查询Device内存利用率为预留功能,当前版本不支持,若调用本接口查询内存利用率,查询到的利用率为-1。 |
父主题: Device管理
在线提单