已关闭
[Requirement|需求建议]: support aclnnBatchNormElemt on Ascend 950 #2719
qianzehong创建于  5月15日关闭于  6月25日
qianzehong成员
5月15日 创建

Backgroud(背景信息)

aclnnBatchNormElemt 当前需要支持 Ascend 950PR/Ascend 950DT 产品。PR 中已将 SocVersion::ASCEND950 纳入 dtype check 的 BF16 支持平台判断,并更新接口文档产品支持表。

Origin(信息来源)

ops-nn 仓 batch_norm_elemt 接口适配需求。

Benefit / Necessity (价值/作用)

支持 Ascend 950PR/Ascend 950DT 上调用 aclnnBatchNormElemt,并覆盖 FLOAT、FLOAT16、BFLOAT16 dtype 校验路径。

Design(设计方案)

  • host 侧 dtype check 中将 SocVersion::ASCEND950 纳入 BF16 支持平台,复用现有 ASCEND910B_DTYPE_SUPPORT_LIST
  • 文档 norm/batch_norm_elemt/docs/aclnnBatchNormElemt.md 中将 Ascend 950PR/Ascend 950DT 产品支持状态更新为支持。
  • 已在 950 上编译自定义包并运行 float 示例与 BF16 smoke。

关联 PR:https://gitcode.com/cann/ops-nn/merge_requests/4894

likedislike
Qqianzehong成员
5月15日 添加了label:requirement
oscillatedoscillated成员
5月15日 将 qianzehong 设为负责人
Qqianzehong成员
6月25日 issue状态由 进行中 改变为 已完成
Qqianzehong成员
6月25日 关闭了 issue
CANN-robotCANN-robot成员
6月25日 添加了label:Accepted