Closed
[Bug-Report|缺陷反馈]: LogAddExp 算子缺少 InferDataType 注册 #4646
zhangqijia1Add on  5 days agoClosed at  2 days ago
zhangqijia1Collaborators
5 days ago Create

一、问题描述

math/log_add_exp/op_host/log_add_exp_infershape.cpp 仅注册了 InferShape,缺少 InferDataType 回调,无法通过该注册显式推导输出 y 的数据类型。

算子定义要求 x1、x2、y 的数据类型一致,支持 FLOAT16、FLOAT 和 BF16。

二、环境信息

代码仓库:cann/ops-math,目标分支:master。

三、重现步骤

查看 LogAddExp 的 IMPL_OP_INFERSHAPE 注册,可以确认仅设置了广播形状推导函数,未设置类型推导函数。

四、预期结果

补充并注册 InferDataTypeForLogAddExp,将输入 x1 的类型设置为输出 y 的类型,返回 ge::GRAPH_SUCCESS。

likedislike
Zzhangqijia1Collaborators
5 days ago Added label: bug-report
Zzhangqijia1Collaborators
5 days ago Associated pull request
陈思
陈思Collaborators
5 days ago Comments:

/assign @gcw_SUaZx3UQ

likedislike
CANN-robotCANN-robotCollaborators
5 days ago Assign gcw_SUaZx3UQ as responsible person
CANN-robotCANN-robotCollaborators
2 days ago Close issue
CANN-robotCANN-robotCollaborators
2 days ago changed custom state from 待办的 to 已完成
CANN-robotCANN-robotCollaborators
2 days ago Added label: resolved