Closed
[Bug-Report|缺陷反馈]: LogAddExp 算子缺少 InferDataType 注册 #4646
zhangqijia1Add on 5 days agoClosed at 2 days ago
5 days ago Added label: bug-report
5 days ago Associated pull request
陈思
5 days ago Comments:
5 days ago Comments:
/assign @gcw_SUaZx3UQ


5 days ago Assign gcw_SUaZx3UQ as responsible person
2 days ago Close issue
2 days ago changed custom state from 待办的 to 已完成
2 days ago Added label: resolved
一、问题描述
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。