已关闭
[Requirement|需求建议]: 为 InplaceApplyPowerSign 算子新增 Tiling 阶段的校验逻辑:dim>8时返回FAILED, 所有参数dtype符合约束 #4631
牛康创建于  8月6日关闭于  8月6日
牛康
牛康
8月6日 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

Backgroud(背景信息)

为 InplaceApplyPowerSign 算子的 Tiling 阶段新增了输入校验逻辑,主要包括输入张量的维度上限检查(rank ≤ 8)以及所有 7 个输入之间 dtype 一致性的校验,并添加了两个 GEIR 异常拦截测试文件来覆盖这些校验路径。

主要改动:
1、新增 CheckInputRankLimit 函数:在 inplace_apply_power_sign_tiling.cpp 中增加了对单个输入的 rank 检查,若维度数超过 8 则返回 GRAPH_FAILED,用于拦截 9 维及以上张量。
2、在 GetShapeAttrsInfo 中加入 rank 和 dtype 一致性校验:循环检查全部 7 个输入(var、m、lr、logbase、sign_decay、beta、grad)的 rank 是否超限,并增加循环校验所有输入的 dtype 是否与 var 的 dtype 一致,不一致则报错返回。

Origin(信息来源)

长尾算子

Benefit / Necessity (价值/作用)

Design(设计方案)

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