GetModelDistributeDesc
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
√ |
|
x |
|
√ |
|
√ |
头文件/库文件
- 头文件：/include/ge/ge_api.h
- 库文件：libge_runner.so
功能说明
查询用于分布式编译部署的模型切分信息。
函数原型
1 | GE_FUNC_VISIBILITY Status GetModelDistributeDesc(const void *data, const uint64_t length, ModelDistibuteDesc &model_dist_desc) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
data |
输入 |
分布式编译部署模型在内存中的地址指针。 |
length |
输入 |
分布式编译部署模型在内存中的字节长度。 |
model_dist_desc |
输出 |
类型为ModelDistibuteDesc，输入的被查询的分布式编译部署模型中分布式编译部署的相关信息（当前只返回部署在几台设备上，后面根据需求扩充，如增加rank table相关信息）。 |
返回值说明
参数名 |
类型 |
描述 |
|---|---|---|
Status |
uint32_t |
接口执行结果。 0：成功。 非0：失败。 |
约束说明
无
父主题： Session API