已关闭
[Requirement|需求建议]: 支持 CosineEmbeddingLoss Ascend 950 #4473
daiwei18创建于  7月31日关闭于  8月3日
daiwei18
daiwei18
7月31日 创建

背景

930 批次需要为 CosineEmbeddingLoss 补齐 arch35/Ascend 950 支持,并通过独立 PR 合入 cann/ops-nn:master

需求范围

  • 为 CosineEmbeddingLoss 增加 arch35/Ascend 950 适配,支持 none、sum 和 mean 归约方式。
  • 补充 arch35 tiling、kernel、Ascend950 binary 配置、infershape、golden 和 UT。
  • 补充算子定义、图原型和 GEIR 示例,覆盖 FP16、FP32、INT32 数据类型。
  • 同步 README、CMake 接入和中文算子清单,保证 loss 算子提交内容完整。

关联提交

验收建议

  • 新增 CosineEmbeddingLoss arch35 op_host/op_kernel UT。
  • 本地执行 pre-commit run,格式、静态检查和 OAT 检查通过。
  • 执行 bash build.sh -u --ops=cosine_embedding_loss --soc=ascend950 -j16,op_host 与 op_kernel UT 通过。
  • 执行 bash build.sh --opkernel --ops=cosine_embedding_loss --soc=ascend950 -j16,Ascend950 binary 编译通过。
  • 精度、泛化和性能结果随转测补充。
likedislike
yuning_chenyuning_chen成员
7月31日 将 daiwei18 设为负责人
CANN-robotCANN-robot成员
8月3日 关闭了 issue
CANN-robotCANN-robot成员
8月3日 添加了label:resolved