| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
Co-authored-by: Lizexin312<lizexin2@huawei.com> # message auto-generated for no-merge-commit merge: !1351 merge lzx_logsoftmax into master LogSoftmaxGrad算子支持Ascend950实现 Created-by: Lizexin312 Commit-by: Lizexin312 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> LogSoftmaxGrad算子支持Ascend950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> [#910](https://gitcode.com/cann/ops-nn/issues/910) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ttk验证/本地example验证 ## 测试用例 std::vector<int64_t> gradOutputShape = {4, 2}; std::vector<int64_t> outputShape = {4, 2}; std::vector<int64_t> outShape = {4, 2}; void* gradOutputDeviceAddr = nullptr; void* outputDeviceAddr = nullptr; void* outDeviceAddr = nullptr; aclTensor* gradOutput = nullptr; aclTensor* output = nullptr; aclTensor* out = nullptr; std::vector<float> gradOutputHostData = {0, 1, 2, 3, 4, 5, 6, 7}; std::vector<float> outputHostData = {1, 1, 1, 2, 2, 2, 3, 3}; std::vector<float> outHostData = {0, 0, 0, 0, 0, 0, 0, 0}; ## 测试结果 result[0] is: -32.619381 result[1] is: -42.492508 result[2] is: -30.619381 result[3] is: -115.224899 result[4] is: -84.6686 result[5] is: -113.224899 result[6] is: -235.026443 result[7] is: -314.368591 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1351 | 5 个月前 |