| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[CANNBot]MaxPoolV3Grad算子支持Ascend950确定性实现 Co-authored-by: m0_46386992<wanghongbin19@huawei.com> # message auto-generated for no-merge-commit merge: !8301 merge add_max_pool_v3_grad into master [CANNBot]MaxPoolV3Grad算子支持Ascend950确定性实现 Created-by: m0_46386992 Commit-by: m0_46386992 Merged-by: cann-robot Description: ## 描述 MaxPoolV3Grad算子在非确定路径下经过测试,相较于确定性实现无明显优势。 通过ForwardPass -> SyncAll -> GatherPass来消除不确定问题,能够减少一次SyncAll,也能够减少workSpace的消耗。 ## 关联的issue https://gitcode.com/cann/ops-nn/issues/4467 ## 测试 - 编译验证通过(ascend950 --pkg) - 三方比对测试通过(NPU vs CPU vs GPU) - 无TF通路,无Pytorch通路 - 确定性实现 - GEIR通路验证通过 ## 文档更新 - SE 需求设计文档 - MDE 详细设计文档 - spec.yaml 交付件 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述 ## AI/Agent 生成声明 代码来源: - [ ] 纯人工手写 - [ ] AI 辅助编写 - [x] AI 完全生成 人工审查:已完成 测试验证:已通过 合规检查:已完成 Assisted-by:CANNBot: glm-5.2 See merge request: cann/ops-nn!8301 | 2 个月前 | |
增加max_pool_v3_grad实现 Co-authored-by: m0_46386992<wanghongbin19@huawei.com> # message auto-generated for no-merge-commit merge: !7715 merge whb_max_pool_v3_grad into master 增加max_pool_v3_grad实现 Created-by: m0_46386992 Commit-by: m0_46386992 Merged-by: cann-robot Description: ## 描述 MaxPoolV3Grad算子支持Ascend950 MaxPoolV3Grad是MaxPoolV3(最大池化)算子的反向传播(梯度计算)算子,用于在神经网络训练过程中将损失函数对池化输出的梯度回传到池化输入。对于每个池化输出位置,其梯度仅传递给前向计算中产生该最大值的输入位置(argmax位置),其余输入位置的梯度为0。采用first-wins策略,当窗口内有多个输入位置等于最大值时,仅首个匹配位置接收梯度。 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/4429 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 编译验证通过(ascend950 --pkg) - 三方比对测试通过(NPU vs CPU vs GPU) - 无TF通路,无Pytorch通路 - 窗口不重叠时,为确定性实现;窗口重叠时,使用atomicAdd原子加,存在因累加顺序误差导致的不确定性现象 - GEIR通路验证通过 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> - 新增README.md - 新增op_graph原型定义 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7715 | 2 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 2 个月前 | ||
| 2 个月前 |