已关闭
[Requirement|需求建议]: segsum/exp_segsum_grad op_kernel头文件按arch22/arch35重命名 #3152
songkai111创建于 9月12日关闭于 9月14日
9月12日 添加了label:requirement
songkai111
9月12日 评论:
9月12日 评论:
/assign @songkai111


9月12日 将 songkai111 设为负责人
9月14日 关闭了 issue
9月14日 添加了label:resolved
一、背景信息 (必填)
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.hmath/exp_segsum_grad/op_kernel/arch35/exp_segsum_grad.h→exp_segsum_grad_arch35.hmath/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