变更摘要

本 PR 为 Atlas A2（Ascend910B / DAV_2201）平台新增 AcosGradV2 算子，实现前向 Acos 的反向梯度计算，公式为 z = -dy / sqrt(1 - y^2)。整个算子完全新增于 experimental/math/acos_grad_v2/ 目录下，包含标准的 ACLNN 注册-调用工程结构（op_host / op_kernel / op_api / aclnn），支持 FP32 / FP16 / BF16 三种数据类型和 ND 格式，仅注册 ascend910b 平台。Kernel 采用 Ascend C 向量 API 实现，FP16/BF16 路径先转为 FP32 计算再转回，FP32 路径直接计算。

主要改动

• 算子定义与注册：在 op_host/acos_grad_v2_def.cpp 中定义 AcosGradV2 OpDef，声明 y（前向输入）、dy（上游梯度）、z（输出梯度）三个张量的数据类型（FP16/FP32/BF16）、格式（ND）及 AiCore 配置，仅通过 AddConfig("ascend910b", ...) 注册 A2 平台。

• 形状与类型推导：op_host/acos_grad_v2_infershape.cpp 实现 InferShape4AcosGradV2，将输出 z 的 shape 直接设为输入 y 的 shape，确保三者 shape 一致。

• Tiling 切分逻辑：op_host/arch32/acos_grad_v2_tiling.cpp 实现 AcosGradV2TilingFunc，根据 UB 大小（184KB）、元素对齐（512）、数据类型等信息计算 block 数量、UB 分块大小等参数，写入 AcosGradV2TilingData 结构体并通过 ASCENDC_TPL_SEL_PARAM 按 dtype 选择模板实例。

• Ascend C Kernel 实现：op_kernel/arch32/acos_grad_v2.h 提供泛型模板 KernelAcosGradV2<T>（FP16/BF16：Cast→FP32 计算→Cast 回原类型）和 float 全特化版本（直接 FP32 计算，省去 Cast），核心计算流程为 Mul→Muls(-1)→Adds(1)→Sqrt→Muls(dy, -1)→Div，通过双缓冲 TQue 和 PipeBarrier 实现流水线并行。

• ACLNN L0/L2 API：op_api/acos_grad_v2.cpp 实现 l0op::AcosGradV2()（形状推导→平台/dtype 校验→AiCore 调度），op_api/aclnn_acos_grad_v2.cpp 实现 aclnnAcosGradV2GetWorkspaceSize 和 aclnnAcosGradV2（参数校验→Contiguous→ViewCopy→计算执行），对外暴露标准 ACLNN 接口。

代码审查

审查总结

各文件审查结果：

整体风险评估：中低风险。 三个 P2 问题均不会在正常执行路径上触发（AllocTensor 失败极为罕见、示例代码的错误路径在实际运行中不会走到、缺少 PipeBarrier 的 FP32 路径当前精度测试全通过），但建议修复以提升代码健壮性和一致性。算子核心计算逻辑（tiling 切分、kernel 公式实现）正确，精度与性能验证均达标。

💬 仅评论

/check-pr

The MR can not be merged, because of CodeReview discussion not resolved

If you want to solve this problem, you can click here to do it in the FAQs.