Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
为 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 一致,不一致则报错返回。
长尾算子
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(设计方案)