930 批次需要为 CosineEmbeddingLoss 补齐 arch35/Ascend 950 支持,并通过独立 PR 合入 cann/ops-nn:master。
CosineEmbeddingLoss
cann/ops-nn:master
daiwei18:cosine
pre-commit run
bash build.sh -u --ops=cosine_embedding_loss --soc=ascend950 -j16
bash build.sh --opkernel --ops=cosine_embedding_loss --soc=ascend950 -j16
背景
930 批次需要为
CosineEmbeddingLoss补齐 arch35/Ascend 950 支持,并通过独立 PR 合入cann/ops-nn:master。需求范围
关联提交
daiwei18:cosine验收建议
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 编译通过。