昇腾社区首页
中文
注册

函数:get_dynamic_hw

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

根据模型描述信息获取模型支持的动态宽高信息。

函数原型

  • C函数原型
    1
    aclError aclmdlGetDynamicHW(const aclmdlDesc *modelDesc, size_t index, aclmdlHW *hw)
    
  • python函数
    1
    hw_info, ret = acl.mdl.get_dynamic_hw(model_desc, index)
    

参数说明

参数名

说明

model_desc

int,aclmdlDesc类型数据的指针地址。

需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。

index

int,预留参数,当前未使用,固定设置为-1。

返回值说明

返回值

说明

hw_info

dict,获取的模型支持的宽高信息。

  • 最大的HW数量为128。
    hw_info = {
                "hwCount": int,#模型中支持的宽高分档数
                "hw":[xx, xx,]  # 模型中支持的具体分档,每组分档中,数组下标为0代表的是高,下标为1代表的是宽
    }
  • “hwCount”等于0时,表示不支持设置档位信息,以模型中的档位为准。

ret

int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。