已关闭
ReduceLogSum 算子缺少 InferDataType 实现 #2626
clinglai0517创建于  28 天前关闭于  25 天前
clinglai0517成员
28 天前 创建

背景

ReduceLogSum 算子已注册 InferShape(基于 axes 张量做归约推导,支持 keep_dims、动态 shape),但未注册 InferDataType,输出 y 的 dtype 无法在构图阶段推导。

影响

  • 构图/图编译阶段无法推导输出 y 的 dtype,可能导致构图失败或运行时错误;
  • 与仓库内其它 reduce 系算子(如 ReduceLogSumExp)不一致。

期望

  1. 补充 InferDataType:y.dtype 跟随输入 x(float / float16 / bf16);
  2. 补充 op_host UT,覆盖 float / float16 / bf16 dtype 场景。

验收标准

在图模式下,能正常推导dtype

likedislike
Cclinglai0517成员
28 天前 关联了pull request:feat: reduce_log_sum补充inferdtype实现及UT用例
Cclinglai0517成员
28 天前 修改标题为 “ReduceLogSum 算子缺少 InferDataType 实现”,原标题为“【社区任务】ReduceLogSum 算子缺少 InferDataType 实现”
Cclinglai0517成员
28 天前 修改了issue 的描述
陈思
陈思成员
27 天前 评论:

/assign @clinglai0517

likedislike
CANN-robotCANN-robot成员
27 天前 将 clinglai0517 设为负责人
CANN-robotCANN-robot成员
25 天前 关闭了 issue
CANN-robotCANN-robot成员
25 天前 添加了label:resolved