对 arch35 架构下 conversion/ 与 math/ 共 15 个算子进行三类整改:
__local_mem__
Reg::DataCopy
Reg::MaskNot
Reg::MaskAnd
__ubuf__
Reg::LoadAlign
Reg::StoreAlign
Reg::Not
Reg::And
TilingForReleatedTranspose
isReleatedTranspsoe_
TilingForRelatedTranspose
isRelatedTranspose_
INFO
only support
is: %s and %s
expect %f
multiples's
涉及算子:broadcast_to、confusion_transpose_d、depth_to_space、matrix_set_diag_v2、pad_v3、pad_v3_grad、pad_v3_grad_replication、space_to_depth、transpose、erfc、get_shape、histogram_fixed_width、sinc、tile、truncate_div。
本次改动不涉及使能方式变更。改动分布于 arch35 kernel(op_kernel/arch35/.h)、host 侧 tiling/infershape/op_api(op_host/arch35/.cpp、op_host/.cpp、op_api/.cpp)及示例代码(examples/*.cpp)。
本次改动不涉及数据类型变更,各算子原有数据类型支持不变。
only supports
are: %s and %s
expected %f
multiples'
arch35 kernel(.h)逐文件替换:
using AscendC::Reg::DataCopy;
涉及算子的硬件支持(依据各算子 CMakeLists.txt 的 COMPUTE_UNIT 字段):
本次改动不涉及算子约束变更,仅迁移 API、修正命名与日志文案。
关联 PR:https://gitcode.com/cann/ops-math/pull/5135
/assign @esok11
一、背景信息 (必填)
对 arch35 架构下 conversion/ 与 math/ 共 15 个算子进行三类整改:
__local_mem__地址空间修饰符和Reg::DataCopy/Reg::MaskNot/Reg::MaskAnd等 API,需迁移至新的__ubuf__、Reg::LoadAlign/Reg::StoreAlign、Reg::Not/Reg::And。TilingForReleatedTranspose(Releated 拼写错误)、isReleatedTranspsoe_(Releated + Transpsoe 拼写错误)需修正为TilingForRelatedTranspose、isRelatedTranspose_。INFO级别、文案存在语法错误(only support、is: %s and %s、expect %f、multiples's)。涉及算子:broadcast_to、confusion_transpose_d、depth_to_space、matrix_set_diag_v2、pad_v3、pad_v3_grad、pad_v3_grad_replication、space_to_depth、transpose、erfc、get_shape、histogram_fixed_width、sinc、tile、truncate_div。
二、价值/作用 (必填)
三、设计方案 (必填)
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
本次改动不涉及使能方式变更。改动分布于 arch35 kernel(op_kernel/arch35/.h)、host 侧 tiling/infershape/op_api(op_host/arch35/.cpp、op_host/.cpp、op_api/.cpp)及示例代码(examples/*.cpp)。
3.2 总体设计
3.2.1 算子支持的数据类型
本次改动不涉及数据类型变更,各算子原有数据类型支持不变。
3.2.2 host侧设计
TilingForReleatedTranspose→TilingForRelatedTranspose、isReleatedTranspsoe_→isRelatedTranspose_,同步 confusion_transpose_d、depth_to_space 中的调用点。only support→only supports、is: %s and %s→are: %s and %s、expect %f→expected %f、multiples's→multiples'。3.2.3 kernel侧设计
arch35 kernel(.h)逐文件替换:
__local_mem__→__ubuf__Reg::DataCopy(加载语义) →Reg::LoadAlignReg::DataCopy(存储语义) →Reg::StoreAlignReg::MaskNot→Reg::NotReg::MaskAnd→Reg::Andusing AscendC::Reg::DataCopy;不改变任何控制流和计算逻辑。
3.3 支持硬件
涉及算子的硬件支持(依据各算子 CMakeLists.txt 的 COMPUTE_UNIT 字段):
3.4 算子约束限制
本次改动不涉及算子约束变更,仅迁移 API、修正命名与日志文案。
关联 PR:https://gitcode.com/cann/ops-math/pull/5135