0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
cann-robotcann-robot修复 AddInputTd 中 malloc 失败后空指针传递问题
578d1d46创建于 24 天前历史提交
文件最后提交记录最后更新时间
context_builder.cpp
修复输入张量大小计算问题 Co-authored-by: zhangyike<zhangyike5@huawei.com> # message auto-generated for no-merge-commit merge: !1146 merge 324-ctx into master 修复输入张量大小计算问题 Created-by: zhangyike Commit-by: zhangyike Merged-by: cann-robot Description: ## 描述 修复了 ContextBuilder 在构建 Tiling Context 时的输入张量大小计算问题。 1. context_builder_impl.cpp 修改 问题： BuildTilingContext() 函数中，tensorValueVec 和 tensorVec 的大小被错误地设置为 inputNum_ + outputNum_，导致：数组索引计算越界;后续GetPlatFormInfo接口获取地址偏移。 修改后输出张量已由 tensorValueVecOut 单独处理。 ## 关联的Issue 关联Issue [#324](https://gitcode.com/cann/asc-devkit/issues/324) ## 测试 单元测试、二级冒烟 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] 🐛 fix: Bug 修复 - [ ] ✨ feat: 新功能 - [ ] ⚡ perf: 性能优化 - [ ] ♻️ refactor: 代码重构 - [ ] 🧪 test: 新增或修改测试 - [ ] 📝 docs: 文档更新 - [ ] 🔧 ci: CI/CD 配置修改 - [ ] ↩️ revert: 回退 - [ ] 🧹 chore: 其他，请具体描述 See merge request: cann/asc-devkit!11462 个月前
context_builder_impl.cpp
修复 AddInputTd 中 malloc 失败后空指针传递问题 Co-authored-by: zhangyike<zhangyike5@huawei.com> # message auto-generated for no-merge-commit merge: !1804 merge 425-issue into master 修复 AddInputTd 中 malloc 失败后空指针传递问题 Created-by: zhangyike Commit-by: zhangyike Merged-by: cann-robot Description: ## 描述 问题描述： 在 AddInputTd 函数中，当 malloc 分配缓冲区失败返回 nullptr 时，仅记录了错误日志， 未进行错误返回或终止流程，导致空指针 buffer 继续被传入 ReadBinFile 函数。 ReadBinFile 内部未对空指针做防护，直接使用 inFile.read 向空地址写入数据，触发未定义行为（空指针解引用）， 修复方案： 在检测到 buffer == nullptr 后，立即记录错误日志并返回，不再继续执行文件读取操作。 同时确保 errFlag_ 被设置，避免后续使用未初始化的数据。 修改文件： impl/utils/context/context_builder_impl.cpp 函数：context_ascendc::ContextBuilderImpl::AddInputTd ## 关联的Issue https://gitcode.com/cann/asc-devkit/issues/529?ref=&did=7b19a715373e58b7b97f091908eba9be4d7fa1e2 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] 🐛 fix: Bug 修复 - [ ] ✨ feat: 新功能 - [ ] ⚡ perf: 性能优化 - [ ] ♻️ refactor: 代码重构 - [ ] 🧪 test: 新增或修改测试 - [ ] 📝 docs: 文档更新 - [ ] 🔧 ci: CI/CD 配置修改 - [ ] ↩️ revert: 回退 - [ ] 🧹 chore: 其他，请具体描述 See merge request: cann/asc-devkit!180424 天前