已合并
refactor: 从block_mem_assigner抽取零拷贝及辅助函数到独立文件 #4229
refactor: 从block_mem_assigner抽取零拷贝及辅助函数到独立文件 #4229
已合并
tangqunzhang创建于 7月31日
tangqunzhang成员
7月31日

Pull Request

描述

从block_mem_assigner.cc中抽取零拷贝判断、内存块生命周期管理、复用策略等函数到独立文件,降低block_mem_assigner.cc复杂度。

抽取的文件:

  • block_mem_zero_copy.cc/h:零拷贝判断相关函数(IsNodeSupportZeroCopy、IsNodeAndPeerNodeTaskSupportZeroCopy等)
  • block_mem_stream.cc/h:流相关函数
  • memory_block.cc/h:内存块管理函数
  • mem_reuse_strategy.cc/h:复用策略函数
  • dynamic_batch_mem_assigner.cc/h:动态batch相关函数

同时补充HCCL算子零拷贝地址刷新判断。

变更类型

关联的Issue

如何测试

  1. 基于 develop 分支编译 ge_compiler
  2. 运行 block_mem_assigner_unittest 单元测试
  3. 确认 IsZeroCopyBlock 相关测试全部通过

核对清单

其他信息

与 PR #4171(抽取AtomicMemoryAssigner)拆分,本PR仅包含block_mem_assigner.cc的函数拆分,不涉及atomic内存分配逻辑。

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 tangqunzhang 的贡献)
Ttangqunzhang成员
7月31日 创建了 pull request,commit 9dde5e32
atomgit-bot
atomgit-bot
7月31日 评论:

本次 AI 摘要未能生成有效摘要。

likedislike
不准确?
atomgit-bot
atomgit-bot
7月31日 评论:

🤖 AI Code Review

⚠️ 审查未能完成

本次代码审查未能完成,可能是临时性故障。

你可以在评论区输入 /ai review 重新发起审查。若多次失败,请联系管理员并附上下方追踪 ID。

追踪 ID: task-142314-run-136493

likedislike
不准确?
CANN-robotCANN-robot成员
7月31日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
7月31日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here


PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
repo-cann/ge ✅ zhanj, kobemini, stevenaw0 (3/2) ✅ zhanj (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

tangqunzhang, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
此处折叠了77条消息 查看更多
CANN-robotCANN-robot成员
7月31日 删除了label:ci-pipeline-running
CANN-robotCANN-robot成员
7月31日 添加了label:ci-pipeline-passed
zhanj
zhanj成员
7月31日 评论:

/approve

likedislike
CANN-robotCANN-robot成员
7月31日 添加了label:approved
CANN-robotCANN-robot成员
7月31日 合入了pull request