Backgroud(背景信息) 为 25 个激活/量化类算子增加 ascend350(arch35)平台支持,使其能在 ascend350 芯片上编译与运行;同时将 6 个 aclnn 接口(aclnnRReluWithNoise/aclnnSigmoidBackward/aclnnSwishBackward/aclnnGeluV2/aclnnBinaryCrossEntropyWithLogitsTargetBackward/aclnnLogSigmoidBackward)的 SocVersion 平台判断重构为 NpuArch 判断。
覆盖算子:AscendQuant/AscendAntiQuant/Quantize、Elu/EluGrad/EluGradV2、FastGelu/FastGeluGrad、Gelu/GeluGrad/GeluV2/GeluGradV2、ReluGrad/ReluGradV2/ReluV2、LeakyReluGrad、SigmoidGrad/SiluGrad、Swish/SwishGrad、HardtanhGrad、Mish/MishGrad、LogSigmoid/LogSigmoidGrad。
Origin(信息来源) CANN ops-nn 算子库存量算子 ascend350 平台适配任务。
Benefit / Necessity(价值/作用) 补齐激活与量化类算子在 ascend350 平台的适配,复用 arch35 tiling/kernel 实现,扩展算子在昇腾新芯片上的可用性;NpuArch 判断使平台分支逻辑与芯片架构代际解耦,便于后续新平台扩展。
Design(设计方案)
关联 PR:https://gitcode.com/cann/ops-nn/merge_requests/9834
/assign @wenqi_wu
Backgroud(背景信息)
为 25 个激活/量化类算子增加 ascend350(arch35)平台支持,使其能在 ascend350 芯片上编译与运行;同时将 6 个 aclnn 接口(aclnnRReluWithNoise/aclnnSigmoidBackward/aclnnSwishBackward/aclnnGeluV2/aclnnBinaryCrossEntropyWithLogitsTargetBackward/aclnnLogSigmoidBackward)的 SocVersion 平台判断重构为 NpuArch 判断。
覆盖算子:AscendQuant/AscendAntiQuant/Quantize、Elu/EluGrad/EluGradV2、FastGelu/FastGeluGrad、Gelu/GeluGrad/GeluV2/GeluGradV2、ReluGrad/ReluGradV2/ReluV2、LeakyReluGrad、SigmoidGrad/SiluGrad、Swish/SwishGrad、HardtanhGrad、Mish/MishGrad、LogSigmoid/LogSigmoidGrad。
Origin(信息来源)
CANN ops-nn 算子库存量算子 ascend350 平台适配任务。
Benefit / Necessity(价值/作用)
补齐激活与量化类算子在 ascend350 平台的适配,复用 arch35 tiling/kernel 实现,扩展算子在昇腾新芯片上的可用性;NpuArch 判断使平台分支逻辑与芯片架构代际解耦,便于后续新平台扩展。
Design(设计方案)
关联 PR:https://gitcode.com/cann/ops-nn/merge_requests/9834