aclrtRegDeviceStateCallback

Applicable Products

Product	Supported
Ascend 950PR/Ascend 950DT	√
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

Function

Registers a device status callback function. Repeated registration is not supported.

When the device status changes (for example, after aclrtSetDevice or aclrtResetDevice is called), the Runtime module triggers the callback function.

Prototype

aclError aclrtRegDeviceStateCallback(const char *regName, aclrtDeviceStateCallback callback, void *args)

Parameters

Parameter	Input/Output	Description
regName	Input	Registered name, which must be unique and cannot be empty. Ensure that the input string ends with \0.
callback	Input	Callback function. If callback is not NULL, the callback function is registered. If callback is NULL, the callback function is deregistered. Prototype: typedef enum { ACL_RT_DEVICE_STATE_SET_PRE = 0, // Before the set API (for example, aclrtSetDevice) is called ACL_RT_DEVICE_STATE_SET_POST, // After the set API (for example, aclrtSetDevice) is called ACL_RT_DEVICE_STATE_RESET_PRE, // Before the reset API (for example, aclrtResetDevice) is called ACL_RT_DEVICE_STATE_RESET_POST, // After the reset API (for example, aclrtResetDevice) is called } aclrtDeviceState; typedef void (aclrtDeviceStateCallback)(uint32_t devId, aclrtDeviceState state, void args);
args	Input	Pointer to the user data to be passed to the callback function.

Parameter

Input/Output

Description

regName

Input

Registered name, which must be unique and cannot be empty. Ensure that the input string ends with \0.

callback

Input

Callback function. If callback is not NULL, the callback function is registered. If callback is NULL, the callback function is deregistered.

Prototype:

typedef enum {
    ACL_RT_DEVICE_STATE_SET_PRE = 0, // Before the set API (for example, aclrtSetDevice) is called
    ACL_RT_DEVICE_STATE_SET_POST,    // After the set API (for example, aclrtSetDevice) is called
    ACL_RT_DEVICE_STATE_RESET_PRE,   // Before the reset API (for example, aclrtResetDevice) is called
    ACL_RT_DEVICE_STATE_RESET_POST, // After the reset API (for example, aclrtResetDevice) is called
} aclrtDeviceState;
typedef void (*aclrtDeviceStateCallback)(uint32_t devId, aclrtDeviceState state, void *args);

args

Input

Pointer to the user data to be passed to the callback function.

Returns

0 on success; otherwise, failure. For details, see aclError.

Parent topic: Device Management