变更摘要

本次 PR 为 CdistGrad 算子补齐了 RT2.0 的输出数据类型推导能力：在 math/cdist_grad/op_host/cdist_grad_infershape.cpp 中新增 InferDataType4CdistGrad，通过算子注册链将其与 InferShape4CdistGrad 一并挂载到 IMPL_OP_INFERSHAPE(CdistGrad) 上；同时在 math/cdist_grad/tests/ut/op_host/test_cdist_grad_infershape.cpp 中补充针对该推导逻辑的 op_host UT 看护。

主要改动

• 新增 dtype 推导函数: 新增 InferDataType4CdistGrad，从 grad 输入读取 dtype 并设置到输出 y，与「grad/x1/x2/cdist 同 dtype」的约束保持一致，并补充起止 OP_LOGD 日志。
• 常量索引定义: 新增 CDIST_GRAD_INPUT_GRAD_INDEX 与 CDIST_GRAD_OUTPUT_Y_INDEX 两个 size_t 常量，替代代码中的魔法数字，用于标识输入 grad 与输出 y 的位置。
• 算子注册链调整: 将注册语句由仅 InferShape 扩展为 IMPL_OP_INFERSHAPE(CdistGrad).InferShape(InferShape4CdistGrad).InferDataType(InferDataType4CdistGrad)，使算子同时具备 shape 与 dtype 推导入口。
• 新增 InferDataType UT: 在 UT 中通过 DefaultOpImplSpaceRegistryV2 获取 CdistGrad 的 opImpl，校验 infer_datatype 存在，并用 OpInferDataTypeContextBuilder 构造 4 输入 1 输出的上下文，覆盖 float32、float16、bfloat16 三种一致 dtype 场景。
• 输入 dtype 不一致场景看护: 新增用例将输入 0 设为 DT_FLOAT、其余输入设为 DT_FLOAT16，验证输出 dtype 仍取 grad 的 DT_FLOAT，并断言返回 ge::GRAPH_SUCCESS。

🧩 本次 Review 已使用自定义 Skill：

• ascendc-code-review

代码审查

✅ 未发现问题，不过下面列了 1 处不太确定的地方，供你参考

🔍 另有 1 处，仅供参考（不影响合并）

下面这些地方 AI 觉得有点可疑，但把握不大，所以没有当成正式问题，也没有在代码行上留评论。列在这里只是多给你一个视角：觉得有道理可以顺手看看，觉得是误报直接忽略就行。

• math/cdist_grad/op_host/cdist_grad_infershape.cpp:80 — suggest: 检查 SetOutputDataType 返回值，失败时返回 GRAPH_FAILED

🧩 本次 Review 已使用自定义 Skill：

• ascendc-code-review

Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
For more, you also can visit HICANN.

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

💡 Tip:

• Committer can comment /approve or /lgtm
• Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

VoyageZhou, thanks for your pull request. All authors of the commits have signed the CLA. 👍

compile

/lgtm
/approve
by agent

您好，PR已满足合入指标，请尽快联系Committer进行合入。如暂时不需要合入，请在PR标题前加[WIP]

/lgtm
/approve