GetModelDistributeDesc

Description

Queries model sharding information used for distributed compilation and deployment.

Prototype

GE_FUNC_VISIBILITY Status GetModelDistributeDesc(const void *data, const uint64_t length,ModelDistibuteDesc &model_dist_desc);

Parameters

Parameter

Input/Output

Description

data

Input

Address pointer to a model in the memory for distributed compilation and deployment.

length

Input

Byte length of a model in the memory for distributed compilation and deployment.

model_dist_desc

Output

Information about distributed compilation and deployment of the input model. Currently, only the number of devices where the model is deployed is returned. In later version, more information, such as the rank table information, will be returned.

Returns

Parameter

Type

Description

Status

uint32_t

Execution result.

0: success.

Other values: failure.

Restrictions

None