aclmdlQueryExeOMDesc

Applicability

Product

Supported

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

Description

Obtains the workspace, weight memory, model description information, and memory sizes of static and dynamic shape tasks required for model execution based on the model file.

Prototype

aclError aclmdlQueryExeOMDesc(const char *fileName, aclmdlExeOMDesc *mdlPartitionSize)

Parameters

Parameter

Input/Output

Description

fileName

Input

Pointer to the model file path, which contains the file name. The user who runs the app must have the permission to access the directory.

The model file is an offline model (offline model adapted to the Ascend AI Processor), that is, an *.exeom file.

mdlPartitionSize

Output

Struct pointer to the size of each partition required for model execution. The unit of the partition size is byte.

Returns

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