Closed
修复Yolo算子dtype/format校验缺失问题 #815
wkkk0528Add on  Aug 25Closed at  Aug 25
wkkk0528
Aug 25 Create

修复 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_hwCollaborators
Aug 25 Assign wkkk0528 as responsible person
CANN-robotCANN-robotCollaborators
Aug 25 Close issue
CANN-robotCANN-robotCollaborators
Aug 25 Added label: resolved