已关闭
修复Yolo算子dtype/format校验缺失问题 #815
wkkk0528创建于  8月25日关闭于  8月25日
wkkk0528
8月25日 创建

修复 Yolo 算子(Ascend950)参数校验链缺失问题(仅修改 op_host/yolo_tiling_arch35.cpp 一个文件):

  1. 新增 ValidateDtype:对输入 x 及 3 个输出(coord_data/obj_prob/classes_prob)做 dtype 白名单校验(仅支持 float16/float32)。此前算子无任何 dtype 校验,非法 dtype(如 bfloat16/bool)会穿透至 kernel 链接期才失败;现统一在 tiling 阶段规范拦截(errCode=EZ0021,param_names 精确到具体参数名)。

  2. 新增 ValidateFormat:对输入 x 及 3 个输出的 StorageFormat 做白名单校验(仅支持 ND)。kernel 索引全部按线性 ND 布局计算,FRACTAL_NZ 输入会静默读错数据;现统一在 tiling 阶段规范拦截(errCode=EZ0018)。

likedislike
Ssunhao_hw成员
8月25日 将 wkkk0528 设为负责人
CANN-robotCANN-robot成员
8月25日 关闭了 issue
CANN-robotCANN-robot成员
8月25日 添加了label:resolved