GetModelDistributeDesc

函数功能

查询用于分布式编译部署的模型切分信息。

函数原型

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

参数说明

参数名

输入/输出

描述

data

输入

分布式编译部署模型在内存中的地址指针。

length

输入

分布式编译部署模型在内存中的字节长度。

model_dist_desc

输出

输入的被查询的分布式编译部署模型中分布式编译部署的相关信息(当前只返回部署在几台设备上,后面根据需求扩充,如增加rank table相关信息)。

返回值

参数名

类型

描述

Status

uint32_t

接口执行结果。

0:成功。

非0:失败。

约束说明

无。

产品支持情况

Atlas 推理系列产品(Ascend 310P处理器),支持。

Atlas 训练系列产品,支持。

Atlas A2训练系列产品,支持。

Atlas 200/300/500 推理产品,不支持。

Atlas 200/500 A2推理产品,不支持。