描述

问题

Relu6Grad、LambApplyOptimizerAssign、LambApplyWeightAssign、LambNextMV、LambNextMVWithDecay、LambNextRight、LambUpdateWithLr、LambUpdateWithLrV2、MultilabelMarginLoss、PoissonNllLoss 这 10 个算子在 op_host/*_infershape.cpp 中只注册了 InferShape，没有注册 InferDataType，GE 图通路上输出 dtype 缺少显式推导来源。这 10 个算子的 op_graph 原型、op_host def 与 op_kernel 均齐全，确属有 GE 图通路的算子。

方法

在各算子已有的 op_host/*_infershape.cpp 中新增 InferDataType4<OpType>，并将原有 IMPL_OP_INFERSHAPE 注册扩展为 .InferShape(...).InferDataType(...)。各输出的 dtype 来源按算子语义逐个确定，未套用统一模板：

其中 MultilabelMarginLoss 的 is_target 不能跟随 x，否则图上会把 INT32 的 is_target 推成浮点。各算子 def 中的 dtype 列表按位一一对应（lamb 系列为「全 fp16」或「全 fp32」两种组合，无混精），故以上取源自洽。

关于摆放位置

仓内两种写法都有：以本 PR 基线计，op_host 的 IMPL_OP_INFERSHAPE(...).InferDataType(...) 共 323 个文件，op_graph 的 IMPL_OP(...).InferDataType(...) 共 90 个文件。本次按多数写法放在各算子已有的 op_host infershape 文件内，可直接扩展原注册行，每算子只改一个文件，diff 中仅删除 10 行旧注册行，未触碰存量代码格式。

关联的Issue

关联 Issue #5417

测试

bash build.sh -u --ophost --soc=ascend950 --ops=relu6_grad,lamb_apply_optimizer_assign,lamb_apply_weight_assign,lamb_next_m_v,lamb_next_m_v_with_decay,lamb_next_right,lamb_update_with_lr,lamb_update_with_lr_v2,multilabel_margin_loss,poisson_nll_loss

结果：116 tests PASSED，0 FAILED。其中包含本次为 10 个算子各新增的 1 条 infer_datatype 用例：

[       OK ] relu6grad.relu6grad_infer_datatype
[       OK ] MultilabelMarginLossInferShapeTest.multilabelmarginloss_infer_datatype
[       OK ] PoissonNllLossInfershapeTest.poissonnllloss_infer_datatype
[       OK ] LambApplyOptimizerAssignProtoTest.lambapplyoptimizerassign_infer_datatype
[       OK ] LambApplyWeightAssignProtoTest.lambapplyweightassign_infer_datatype
[       OK ] LambNextMVProtoTest.lambnextmv_infer_datatype
[       OK ] LambNextMVWithDecayProtoTest.lambnextmvwithdecay_infer_datatype
[       OK ] LambNextRightProtoTest.lambnextright_infer_datatype
[       OK ] LambUpdateWithLrProtoTest.lambupdatewithlr_infer_datatype
[       OK ] LambUpdateWithLrV2ProtoTest.lambupdatewithlrv2_infer_datatype

用例中的 ASSERT_NE(dataTypeFunc, nullptr) 实证该注册可被 OpImplRegistry 取到；MultilabelMarginLoss 那条显式断言 is_target 推导为 INT32；Relu6Grad 那条按 def 声明逐 dtype 覆盖 fp16 / fp32 / bf16。

门禁自检：clang-format 18.1.8 --style=file 20 个文件全通过；仓内 scripts/oat_check.sh 20 个文件 All checks passed；trailing-whitespace / end-of-file / 冲突标记检查通过。

未覆盖：本次仅验证到 op_host UT 层（用例通过 OpImplRegistry 取到 infer_datatype 函数指针后直接调用），未做真机与 GE 图 e2e 验证。

文档更新

无。本次仅新增 InferDataType 注册与对应 UT，不涉及对外接口与算子资料变更。

类型标签

• Bug修复
• 新特性
• 性能优化
• 文档更新
• 其他，请描述：

AI/Agent生成声明

• AI辅助编写