[ PASSED ] 47 tests.
[ OK ] FusedMulAddInfershape.infer_datatype_fp16
[ OK ] FusedMulAddInfershape.infer_datatype_fp32
[ OK ] FusedMulAddInfershape.infer_datatype_int32
[ OK ] FusedMulAddAddInfershape.infer_datatype_fp16
[ OK ] FusedMulAddAddInfershape.infer_datatype_fp32
[ OK ] FusedMulAddAddInfershape.infer_datatype_int32
[ OK ] MulNoNanInfershape.infer_datatype_fp16
[ OK ] MulNoNanInfershape.infer_datatype_fp32
[ OK ] MulNoNanInfershape.infer_datatype_bf16
[ OK ] MulNoNanInfershape.infer_datatype_int32
Describe the current behavior / 问题描述
FusedMulAdd、FusedMulAddAdd、MulNoNan 这 3 个算子,在各自的
op_host/*_infershape.cpp中只做了IMPL_OP_INFERSHAPE(<OpType>).InferShape(...)注册,没有注册InferDataType;在op_graph/下也没有IMPL_OP(<OpType>).InferDataType(...)的注册。因此在 GE 图通路上,这 3 个算子的输出 dtype 缺少显式推导来源。Environment / 环境信息
6439d9c90(feat: Add aclnn st cases and golden functions)build.sh -u --ophost)Steps to reproduce the issue / 重现步骤
检出 cann/ops-math master(基线
6439d9c90)对上述任一算子检查注册面,例如:
grep -n "InferDataType" math/fused_mul_add/op_host/fused_mul_add_infershape.cpp grep -rn "InferDataType" math/fused_mul_add/op_graph/两条命令均无输出,即该算子既没有在 op_host 也没有在 op_graph 注册 InferDataType。
math/fused_mul_add_add、math/mul_no_nan同理。infer_datatype字段为 nullptr。Describe the expected behavior / 预期结果
这 3 个算子应注册
InferDataType,输出 dtype 跟随输入 x1(与 op_proto 一致),即从注册表取到的infer_datatype非空,且对 fp16 / fp32 / bf16 / int32 等输入 dtype 均能推导出与 x1 相同的输出 dtype。Related log / screenshot / 日志 / 截图
修复前,上述 grep 命令无任何输出(即无 InferDataType 注册)。
修复后,
bash build.sh -u --ophost --ops=fused_mul_add,fused_mul_add_add,mul_no_nan的执行结果:新增用例中的
ASSERT_NE(opImpl->infer_datatype, nullptr)即为该注册生效的直接证据。Special notes for this issue / 备注