aclmdlExecuteV2

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 Usage

Performs model inference based on the attributes configured using the aclmdlSetExecConfigOpt API until the inference result is returned. This API is enhanced based on aclmdlExecute. It allows you to control configuration parameters during model inference.

This API must be used together with other APIs to implement model execution. The API call sequence is as follows:
  1. Call aclmdlCreateExecConfigHandle to create a configuration object for model execution.
  2. Call aclmdlSetExecConfigOpt multiple times to set the value of each attribute in the configuration object.
  3. Call aclmdlExecuteV2 to specify the configuration information required for model execution and execute the model.
  4. After the model is successfully executed, call aclmdlDestroyExecConfigHandle to destroy the configuration object.

Prototype

1
aclError aclmdlExecuteV2(uint32_t modelId, const aclmdlDataset *input, aclmdlDataset *output, aclrtStream stream, const aclmdlExecConfigHandle *handle)

Parameters

Parameter

Input/Output

Description

modelId

Input

Model ID for inference.

After a model loading API (such as aclmdlLoadFromFile or aclmdlLoadFromMem) is successfully called, the model ID is returned. This ID is used as the input of this API.

input

Input

Pointer to the input data for model inference. For the type definition, see aclmdlDataset.

output

Output

Pointer to the output data for model inference. For the type definition, see aclmdlDataset.

When calling aclCreateDataBuffer to create an aclDataBuffer type for storing the output data of the corresponding index, you can pass nullptr to the data parameter and set size to 0 to create an empty aclDataBuffer type. During model execution, the system automatically calculates and allocates the index output memory. This method saves memory. However, you need to free the memory and reset the aclDataBuffer after using the data. In addition, memory copy is involved when the system allocates memory, which may cause performance loss.

The sample code for freeing the memory and resetting the aclDataBuffer is as follows:
1
2
3
4
aclDataBuffer *dataBuffer = aclmdlGetDatasetBuffer(output, 0); // Obtain the corresponding data buffer based on the index.
void *data = aclGetDataBufferAddr(dataBuffer);  // Obtain the device pointer to data.
aclrtFree(data ); // Free the device memory.
aclUpdateDataBuffer(dataBuffer, nullptr, 0); // Reset the content in the data buffer for next inference.

stream

Input

Stream. For the type definition, see aclrtStream.

handle

Input

Pointer to the configuration object for model execution. For the type definition, see aclmdlExecConfigHandle.

The value must be the same as that of handle in aclmdlSetExecConfigOpt.

Returns

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

Restrictions

If parameter stream is set to NULL, the attribute values configured using the aclmdlSetExecConfigOpt API do not take effect.

Other restrictions are the same as those of aclmdlExecute.