Star0
0
代码
代码
Issues
Pull Requests
Actions
项目讨论
Wiki
分析
Star0
0
cann-robotcann-robotAIDD扫描语义问题修改
ce180288创建于 11 天前历史提交
文件最后提交记录最后更新时间
bidirection_lstm
nn仓rnn 、vfusion、control文件夹打标签 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !8342 merge master into master nn仓rnn 、vfusion、control文件夹打标签 Created-by: yanglu-1 Commit-by: yanglu-1;y60124828 Merged-by: cann-robot Description: ## 描述 nn仓rnn 、vfusion、control文件夹打标签 ## 关联的Issue 例如：关联Issue [#4600](https://gitcode.com/cann/ops-nn/issues/4600) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8342
1 个月前
bidirection_lstmv2
nn仓rnn 、vfusion、control文件夹打标签 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !8342 merge master into master nn仓rnn 、vfusion、control文件夹打标签 Created-by: yanglu-1 Commit-by: yanglu-1;y60124828 Merged-by: cann-robot Description: ## 描述 nn仓rnn 、vfusion、control文件夹打标签 ## 关联的Issue 例如：关联Issue [#4600](https://gitcode.com/cann/ops-nn/issues/4600) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8342
1 个月前
dynamic_rnn
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
11 天前
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
11 天前
gru
fix: 修复 GRU 与 GRU 反向算子 aclnn 接口的编码规范告警 Co-authored-by: Nice_try<zhangzhiwei108@huawei.com> # message auto-generated for no-merge-commit merge: !9180 merge 0825cleancode into master fix: 修复 GRU 与 GRU 反向算子 aclnn 接口的编码规范告警 Created-by: Nice_try Commit-by: Nice_try Merged-by: cann-robot Description: ## 描述 修复 GRU 算子（rnn/gru/op_api/aclnn_gru.cpp）与 GRU 反向算子（rnn/gru_grad/op_api/aclnn_gru_backward.cpp）aclNN 接口的编码规范（cleancode）告警： 1. CheckNotNull、CheckDimsSize 中未使用的 hx 参数，以及无调用点的 GruDataGetWorkspaceSize 函数添加 [[maybe_unused]] 标记，消除未使用告警； 2. CheckShape 中 dScale（uint64_t）与 numLayers、hiddenSize（int64_t）的混合符号运算显式 static_cast 为 int64_t，消除符号转换告警； 3. 修复 aclnn_gru_backward.cpp ValidateShape 中日志格式符错误（%u 改为 %zu，GetDimNum 返回 64 位无符号值），并移除多余的 static_cast 消除符号比较告警。 ## 关联的Issue 关联 Issue https://gitcode.com/cann/ops-nn/issues/5129 ## 测试 通过编码规范（cleancode）检查。 ## 文档更新 无。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-nn!9180
13 天前
gru_grad
fix: 修复 GRU 与 GRU 反向算子 aclnn 接口的编码规范告警 Co-authored-by: Nice_try<zhangzhiwei108@huawei.com> # message auto-generated for no-merge-commit merge: !9180 merge 0825cleancode into master fix: 修复 GRU 与 GRU 反向算子 aclnn 接口的编码规范告警 Created-by: Nice_try Commit-by: Nice_try Merged-by: cann-robot Description: ## 描述 修复 GRU 算子（rnn/gru/op_api/aclnn_gru.cpp）与 GRU 反向算子（rnn/gru_grad/op_api/aclnn_gru_backward.cpp）aclNN 接口的编码规范（cleancode）告警： 1. CheckNotNull、CheckDimsSize 中未使用的 hx 参数，以及无调用点的 GruDataGetWorkspaceSize 函数添加 [[maybe_unused]] 标记，消除未使用告警； 2. CheckShape 中 dScale（uint64_t）与 numLayers、hiddenSize（int64_t）的混合符号运算显式 static_cast 为 int64_t，消除符号转换告警； 3. 修复 aclnn_gru_backward.cpp ValidateShape 中日志格式符错误（%u 改为 %zu，GetDimNum 返回 64 位无符号值），并移除多余的 static_cast 消除符号比较告警。 ## 关联的Issue 关联 Issue https://gitcode.com/cann/ops-nn/issues/5129 ## 测试 通过编码规范（cleancode）检查。 ## 文档更新 无。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-nn!9180
13 天前
single_layer_lstm_grad
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
11 天前
thnn_fused_lstm_cell
AIDD扫描语义问题修改 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !9352 merge master into master AIDD扫描语义问题修改 Created-by: yanglu-1 Commit-by: y60124828 Merged-by: cann-robot Description: ## 描述 AIDD扫描语义问题修改 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/5254 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9352
11 天前
thnn_fused_lstm_cell_grad
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
11 天前
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
9 个月前