已关闭
[Requirement|需求建议]: segsum/exp_segsum_grad op_kernel头文件按arch22/arch35重命名 #3152
songkai111创建于  9月12日关闭于  9月14日
songkai111成员
9月12日 创建

一、背景信息 (必填)

segsum 与 exp_segsum_grad 两个算子的 op_kernel 目录存在同名头文件:arch22(默认)与 arch35 各有一份 exp_segsum_grad.h,segsum.h 也未体现架构归属。同名头文件在编译期按 include 路径解析,容易产生歧义,不利于多架构并行维护。

二、价值/作用 (必填)

通过文件名直接区分头文件所属架构,提升代码可读性与可维护性,避免 arch22/arch35 之间混淆,降低后续适配新架构时的出错风险。

三、设计方案 (必填)

3.1 使能方式

不涉及使能方式变更,仅调整 kernel 侧头文件命名与 include。

3.2 总体设计
3.2.1 算子支持的数据类型

不涉及数据类型变更。

3.2.2 host侧设计

无变更。

3.2.3 kernel侧设计
  • math/exp_segsum_grad/op_kernel/exp_segsum_grad.h → exp_segsum_grad_arch22.h
  • math/exp_segsum_grad/op_kernel/arch35/exp_segsum_grad.h → exp_segsum_grad_arch35.h
  • math/segsum/op_kernel/segsum.h → segsum_arch22.h
  • 同步更新对应 .cpp 的 include 及 include guard。
3.3 支持硬件

arch22、arch35(Ascend 950)等;命名调整不改变实际支持范围。

3.4 算子约束限制

无。

💡 备注(选填)
关联 PR:https://gitcode.com/cann/ops-math/pull/5615

likedislike
Ssongkai111成员
9月12日 添加了label:requirement
songkai111成员
9月12日 评论:

/assign @songkai111

likedislike
CANN-robotCANN-robot成员
9月12日 将 songkai111 设为负责人
CANN-robotCANN-robot成员
9月14日 关闭了 issue
CANN-robotCANN-robot成员
9月14日 添加了label:resolved