tf_plugin

cann-robot修复Gru算子执行报错问题

864ea2ab创建于 8月31日历史提交

文件	最后提交记录	最后更新时间
bidirection_lstm	md大模型检测低错修复 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !9420 merge master into master md大模型检测低错修复 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md aidd大模型检测低错修复 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5254 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> acl*.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9420	1 个月前
bidirection_lstmv2	md大模型检测低错修复 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !9420 merge master into master md大模型检测低错修复 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md aidd大模型检测低错修复 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5254 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> acl*.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9420	1 个月前
dynamic_rnn	md大模型检测低错修复 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !9420 merge master into master md大模型检测低错修复 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md aidd大模型检测低错修复 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5254 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> acl*.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9420	1 个月前
dynamic_rnnv2	refactor: 拆分 RNN/LSTM 系列算子大函数，提升代码可维护性 Co-authored-by: Nice_try<zhangzhiwei108@huawei.com> # message auto-generated for no-merge-commit merge: !9331 merge bigFunc into master refactor: 拆分 RNN/LSTM 系列算子大函数，提升代码可维护性 Created-by: Nice_try Commit-by: Nice_try Merged-by: cann-robot Description: ## 描述 本 PR 对 RNN/LSTM 系列算子（dynamic_rnn、dynamic_rnnv2、single_layer_lstm_grad、thnn_fused_lstm_cell_grad）进行代码重构：将各层（op_api / op_host / op_kernel / framework）中体量过大的函数拆分为多个职责单一的小函数，不改动算子行为与对外接口语义。 ### 主要改动 - **dynamic_rnn**： - framework/npu_lstm_onnx_plugin.cpp：ParseOpToGraphNpuLSTM 拆分为 GetNpuLSTMAttrs、SetNpuLSTMGraphIO - op_api/aclnn_lstm.cpp：aclnnLSTMGetWorkspaceSize 拆分为 PrepareLstmWorkspaceInputs、ProcessTrainLayers、ProcessInferLayers（内部再细分 ProcessTrainLayerForward/Backward、ProcessInferLayerForward/Backward、AllocInferBackwardTensors）；权重/bias/init_hc 预处理抽取 PrepareWeightTrans、PrepareBias、PrepareInitHC；输出处理拆分为 ProcessHyCyOutputs、ProcessTrainOutputs；shape 校验拆分为 CheckWeightShapes、SetDataInfo、CheckParamsShapes、CheckOutputListSize - op_host/dynamic_lstm_tiling.cpp/.h：GetMMTilingDataSplit 拆分为 GetMMInputTiling、GetMMHiddenTiling - op_host/dynamic_rnn_infershape.cpp：shape 推导拆分为 GetInputShapes、GetOutputShapes、GetHiddenSize、SetOutputShapes - op_kernel/LstmFP32.cpp/.h：ProcessVectorOnce/ProcessVectorInitHC 拆分为 CalcVectorBlockSize、ProcessSeqLengthUpdateC/H、ProcessTanhC、ProcessInitSeqLengthC/H - op_kernel/dynamic_rnn_common.h：InitBuffers/InitBuffersV2 拆分为 InitBuffersOffsets、SetInputGmBuffers(V2)、SetOutputGmBuffers - **dynamic_rnnv2**： - framework/npu_lstm_cell_onnx_plugin.cpp：ParseOpToGraphNpuLSTMCell 拆分为 GetNpuLSTMCellAttrs、SetNpuLSTMCellGraphIO - **single_layer_lstm_grad**： - op_api/aclnn_lstm_backward.cpp：双向 LSTM 反向计算拆分为 GetBidirWeightsBias、ExecBidirLayerForward、ExecBidirLayerBackward；权重拼接抽取 ConcatWeightBackward；shape 校验拆分为 ValidateLayerShapes、ValidateGateShapes、CheckGateListLengths、CheckParamsListLength；逐层梯度处理抽取 ProcessLstmGradLayer、ReshapeLstmDataInput - op_host/single_layer_lstm_grad_tiling.cpp：GetMMTilingDataSplit 拆分为 GetMMDgateTiling、GetMMDwTiling；SetTilingData 拆分为 SetScalarTilingData、SetCutBatchTilingData；CheckParamsShape 拆分为 CheckXAndInitHShape、GetOptionalInputFlags、ValidateInputShapes、ValidateOutputShapes - op_kernel/single_layer_lstm_grad.h：InitGlobalBuffers 拆分为 InitGmOffsets、SetInputGmBuffers、SetOutputGmBuffers；LoadDgatesInputData 拆分为 LoadDgatesInputDataFloat、LoadDgatesInputDataNonFloat；ComputeDgates 拆分为 ComputeDgatesDH、ComputeDgatesGrads、ApplyDgatesSeqLength；大循环体拆分为 ProcessDxhSplitTask、ProcessDxSplitLargeNLoop、ProcessCopyInputXLargeNLoop、ProcessCopyHiddenHLargeNLoop 等 - **thnn_fused_lstm_cell_grad**： - op_kernel/thnn_fused_lstm_cell_grad.h：ComputeDgates 拆分为 ComputeDgatesDC、ComputeDgatesGates ## 关联的Issue 关联 Issue：https://gitcode.com/cann/ops-nn/issues/5239 ## 测试 本 PR 为纯重构（函数拆分，行为不变），需回归验证 LSTM 相关用例（动态 shape、双向、训练/推理、seq_length mask 等场景）。 ## 文档更新 无文档更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码重构（拆分大函数） ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-nn!9331	1 个月前
gru	修复Gru算子执行报错问题 Co-authored-by: chenyifeng<chenyifeng27@h-partners.com> # message auto-generated for no-merge-commit merge: !9449 merge gru_0828 into master 修复Gru算子执行报错问题 Created-by: chen_0715 Commit-by: chenyifeng Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复GRU算子执行报错507015的问题，通过对其baseN，解决UB越界报错。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5403 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地自验证200case没问题，问题case验证通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9449	1 个月前
gru_grad	修复gru_grad 编译告警 Co-authored-by: sakuraqqz<furao15@h-partners.com> # message auto-generated for no-merge-commit merge: !9590 merge master into master 修复gru_grad 编译告警 Created-by: sakuraqqz Commit-by: sakuraqqz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复gru_grad 编译告警 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5404 ## 测试 <!--描述进行了哪些测试来验证你的改动。-->  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9590	1 个月前
single_layer_lstm_grad	md大模型检测低错修复 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !9420 merge master into master md大模型检测低错修复 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md aidd大模型检测低错修复 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5254 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> acl*.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9420	1 个月前
thnn_fused_lstm_cell	test: 新增 aclnnThnnFusedLstmCell 算子 ST 测试用例及 CPU golden 执行插件 Co-authored-by: Nice_try<zhangzhiwei108@huawei.com> # message auto-generated for no-merge-commit merge: !9567 merge upload-st into master test: 新增 aclnnThnnFusedLstmCell 算子 ST 测试用例及 CPU golden 执行插件 Created-by: Nice_try Commit-by: Nice_try Merged-by: cann-robot Description: ## 描述 本 PR 为 thnn_fused_lstm_cell 算子新增 ST（系统测试）用例：基于 ATK 测试框架，新增 208 条 aclnnThnnFusedLstmCell 用例配置及对应的 CPU golden 执行插件。 ### 改动内容 - 新增 rnn/thnn_fused_lstm_cell/tests/st/aclnnThnnFusedLstmCell/all_aclnnThnnFusedLstmCell.json：208 条 ST 用例（id 0~207，version v2.1），覆盖 fp16/fp32 两种输入数据类型，shape 覆盖 batch 0~16777216、hidden_size 4~67108864 的常规/大 shape/边界（is_boundary，共 8 条）场景。 - 新增 rnn/thnn_fused_lstm_cell/tests/st/aclnnThnnFusedLstmCell/executor_aclnnThnnFusedLstmCell.py：ATK 执行插件，注册 thnn_fused_lstm_cell_api API 类型，实现算子 CPU golden 计算（gates = inputGates + hiddenGates + b_ih + b_hh，门控激活 i/f/g/o，cy = f*cx + i*g，hy = o*tanh(cy)）；低精度输入先转 fp32 计算再回转，与 NPU 内核 cast-in/fp32 计算/cast-out 行为对齐，保证混合容差比对有效。 ## 关联的Issue - 关联 Issue #5381：https://gitcode.com/cann/ops-nn/issues/5381 ## 测试 - 新增 ST 用例通过 ATK 框架执行：CPU 路径由 executor 插件计算 golden，NPU 路径使用 atk aclnn 命令经 pyaclnn 后端调用两段式 aclnn API（aclnn_name=ThnnFusedLstmCell）。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-nn!9567	1 个月前
thnn_fused_lstm_cell_grad	md大模型检测低错修复 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !9420 merge master into master md大模型检测低错修复 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md aidd大模型检测低错修复 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5254 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> acl*.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9420	1 个月前
CMakeLists.txt	Format update: Adjusted the license description format in all relevant files. Co-authored-by: jiangqi<jiangqi47@h-partners.com> # message auto-generated for no-merge-commit merge: !318 merge license2 into master Format update: Adjusted the license description format in all relevant files. Created-by: jiangqi2025 Commit-by: jiangqi Merged-by: turing_project1 Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!318	10 个月前