青海大学

/compile

变更摘要

本 PR 为 CANN 代码侦探挑战赛（青海大学）提交的 challenge08_tanhcustom 算子工程，使用 AscendC 实现 Tanh 激活算子。host 侧在 tanh_tiling.cpp 中完成按核并行切分的 Tiling（向上取整、尾核截断防御、UB 容量校验并预留 guard 字节），kernel 侧在 tanh.cpp/tanh.h 中按 schMode 分派 fp32/fp16 双实例，通过 Exp/Muls/Add/Sub/Div 计算链实现 y=(e^x-e^-x)/(e^x+e^-x)，搬运统一使用 DataCopyPad；同时配套 aclnn 调用示例、op_host/op_kernel 双层 UT 与基于 MERE/MARE 的精度比对脚本。

主要改动

• host 侧算子定义与 Tiling：tanh_def.cpp 声明 Tanh 算子（DT_FLOAT/DT_FLOAT16、FORMAT_ND），tanh_infershape.cpp 将输入 shape 原样赋给输出，tanh_tiling.cpp 依据 dtype 选择 tileAlignHint、按元素数选取 coreGrain 计算 plannedCores/coreSpan/launchCores，并在 UB 容量扣减 kUbGuardBytes 后计算 ubFactor，最后按输入 dtype 设置 tilingKey（fp16→TANH_TPL_SCH_MODE_0，fp32→TANH_TPL_SCH_MODE_1）。
• kernel 计算与分派：tanh.h 中模板类 NsTanh::Tanh<T> 的 ComputeTanh 以 Exp→Muls→Exp→Add→Sub→Div 顺序实现 tanh，Init 按 blockFactor 计算各核 coreOffset/activeElems_（越界核置 0 空转）并决定 compact/queue 缓冲模式；tanh.cpp 通过 if constexpr 按 schMode 实例化 Tanh<half> 与 Tanh<float>，CopyIn/CopyOut 统一使用 DataCopyPad。
• Tiling 数据结构与模板参数：新增 tanh_tiling_data.h 定义 TanhTilingData（totalNum/blockFactor/ubFactor 三个字段），tanh_tiling_key.h 通过 ASCENDC_TPL_ARGS_DECL 声明 schMode 模板参数及 TANH_TPL_SCH_MODE_0/1 两个取值。
• 示例与构建脚本：examples/test_aclnn_tanh.cpp 演示 aclnnTanhGetWorkspaceSize/aclnnTanh 两段式调用（含 fp16/bf16 数据转换、workspace 申请与资源释放）；顶层 build.sh 提供 -j、-u、-e、--make_clean 选项，串起 cmake 构建、UT 与示例执行。
• UT 与精度比对体系：tests/ut 下提供 gtest 构建（BuildGoogleTest.cmake，强制 old ABI 以匹配 CANN 库）、TilingContextFaker/InferShapeContextFaker 与对应 case executor；test_tanh_tiling.cpp 校验 tiling 输出，test_tanh.cpp 直接构造 TanhTilingData 经 ICPU_RUN_KF((tanh<1>)) 运行 kernel，gen_data.py/compare_data.py 生成 golden 并以 MERE/MARE 阈值（fp16/fp32 等按 dtype 设定）完成精度比对，run.sh 汇总各阶段结果。

代码审查

审查结论

各文件审查状态

• CMakeLists.txt — 已审查。无独立问题（第 29 行 --soc 用法提示与 build.sh 不一致，已并入 build.sh 的问题 3）。
• build.sh — 已审查。2 个问题（P3：--soc 参数未实现但被提示使用；P3：-e 分支在未安装算子包时必然失败）。
• examples/CMakeLists.txt — 已审查。无独立问题（依赖已安装算子包，相关风险并入 build.sh 问题 4）。
• examples/run.sh — 已审查。无独立问题。
• examples/test_aclnn_tanh.cpp — 已审查。1 个问题（P3：示例不回读/校验输出，精度回归无法被发现）。
• op_host/CMakeLists.txt — 已审查。无问题。
• op_host/tanh_def.cpp — 已审查。无问题（AddConfig 仅 ascend910b，与 build.sh 默认 SOC 一致）。
• op_host/tanh_infershape.cpp — 已审查。无问题。
• op_host/tanh_tiling.cpp — 已审查。无问题（tiling key 分派与 kernel 的 fp16/fp32 双实例一致；tile 上限在 192KB UB 下 5 个缓冲区可容纳）。
• op_kernel/CMakeLists.txt — 已审查。无问题。
• op_kernel/tanh.cpp — 已审查。无问题。
• op_kernel/tanh.h — 已审查。2 个问题（P2：大绝对值输入 exp 溢出输出 NaN；P2：尾块 DataCopyPad 非 32B 对齐时越界读写 GM）。

问题统计

• P0：0
• P1：0
• P2：2
• P3：3

总体风险判断

整体构建与基本运行链路完整，host 侧 tiling/def/infershape 及 kernel 分派逻辑一致，未见编译级或接口级阻断问题。主要风险集中在 kernel 数值正确性与边界安全：ComputeTanh 在 |x| 较大（fp32 ≥ ~88.7、fp16 ≥ ~11.1）时因 exp 溢出产生 NaN，属于真实边界正确性缺陷（P2）；输出尾块使用 DataCopyPad 在张量尺寸非 32B 对齐时存在 GM 越界读写（P2）。脚本与示例侧为便利性问题（P3）。建议合入前优先修复 tanh.h 的数值溢出与尾块越界问题，并补充大绝对值输入的校验用例。

审查结论

发现统计

• P0：0 个
• P1：0 个
• P2：2 个
• P3：3 个

总体风险判断

本次 PR 全部为 UT 基础设施与 tiling 数据/模板参数头文件（TanhTilingData、TanhTilingKey），生产算子本体（host tiling / kernel）不在 diff 内。整体风险中低：无 P0/P1 级问题，最值得关注的两点是——gtest 从外部镜像下载编译却关闭 TLS 校验且无校验和（供应链风险），以及 infershape 用例执行器缺少空指针校验（算子未注册时测试进程直接崩溃而非可诊断失败）。其余为条件触发的健壮性问题。

各文件审查确认

1. op_kernel/tanh_tiling_data.h — no issues（结构体三字段 int64 + 默认初始化，无对齐/溢出问题）
2. op_kernel/tanh_tiling_key.h — no issues（标准 ASCENDC_TPL 宏用法，参数列表与模板声明一致）
3. tests/ut/CMakeLists.txt — no issues（UT 接线常规写法；子目录 CMakeLists 不在审查范围内无法交叉验证）
4. tests/ut/cmake/BuildGoogleTest.cmake — P2：TLS_VERIFY OFF 且无 URL_HASH，第三方 gtest 下载无完整性校验（供应链风险）
5. tests/ut/common/any_value.h — no issues（shared_ptr<void> 正确管理 new 对象，各 CreateFrom 特化与枚举类型一一对应）
6. tests/ut/common/infershape_case_executor.cpp — P2：spaceRegistry/GetOpImpl()/infer_shape 无判空即解引用，与 tiling 侧判空不一致，注册缺失时崩溃
7. tests/ut/common/infershape_case_executor.h — no issues
8. tests/ut/common/infershape_context_faker.cpp — no issues（均为 builder 直通调用）
9. tests/ut/common/infershape_context_faker.h — no issues（Attr 重载忽略 attrName 属设计限制，本算子无属性，未达上报标准）
10. tests/ut/common/tiling_case_executor.cpp — P3：STR(BUILD_SOC_VERSION) 字符串化语义使"宏未定义"防御失效，且 socToArch 缺键导致 socVersion/NpuArch 静默为空
11. tests/ut/common/tiling_case_executor.h — P3：5 参重载默认 expectTilingData="" 与跳过哨兵 EMPTY_EXPECT_TILING_DATA 不一致，默认调用产生假失败
12. tests/ut/common/tiling_context_faker.cpp — P3：DeterministicInfo 未判空即解引用指针参数，空指针调用会崩溃

审查结论

变更文件逐一核对结果

汇总

• 发现数：P0 = 0，P1 = 0，P2 = 0，P3 = 6。
• 总体风险评估：本 PR 为 tanh 算子的 UT 测试体系（host tiling UT + kernel UT + 精度比对 + 构建/执行脚本），不涉及生产算子逻辑本身。未发现会导致构建失败、数据损坏或安全风险的缺陷；所报 6 个问题均为测试/脚本健壮性问题（错误路径未校验、异常未捕获、死代码、环境依赖硬编码），触发时多以"崩溃/测试失败"等显式方式暴露，不会静默产生错误结论，整体风险低。其中 tiling_context_faker.h 的 Attr() 丢名字问题属于新测试设施 API 的潜在正确性陷阱，当前用例未触发，建议尽快修正以免后续用例踩坑。

💬 仅评论