已合并
【PR】: aclgraph software sq for ub #2494
【PR】: aclgraph software sq for ub #2494
已合并
zhut66创建于 5月29日
zhut66成员
5月29日

Pull Request

预计合入时间:0609

描述

aclgraph UB 扩流适配
核心目标:为 aclgraph 场景实现 Jetty(异步 DMA 队列)的预申请、池化管理和复用机制,提升图重复执行性能。
三个阶段:

  1. Capture Begin:Stream 首次收到异步拷贝任务时,申请 2k 深度的 normal jetty 放入 Pool;创建 host WQE buffer 存储任务描述符
  2. Capture End:用 NOP WQE 填满未满的 WQE buffer
  3. Graph Reply:从 Pool 分配空闲 jetty 绑定到 stream;同步 WQE buffer 到设备;大深度(>2k)场景临时申请/释放
    关键组件:
  • JettyPool:底层资源池,管理 H2D/D2D 两类 jetty,上限 2k 个
  • JettyManager:流级别管理器,维护 stream→jetty 映射和 Capture 上下文
  • StreamCaptureJettyContext:管理 host WQE buffer 和 SQE 位置记录
  • StreamCaptureJettyHandler:处理 WQE 生成、jetty 绑定、SQE 刷新

变更类型

请选择本次引入的变更类型:

关联的Issue

https://gitcode.com/cann/runtime/issues/600

如何测试

rdv, ub继承扩流用例

核对清单

其他信息

在此添加任何其他关于本次 PR 的说明。

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zhut66 的贡献)
Zzhut66成员
5月29日 添加了label:enhancement
Zzhut66成员
5月29日 创建了 pull request,commit 62c96b3a
CANN-robotCANN-robot成员
5月29日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
5月29日 评论:

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
docs 卢煜坤, yanmingxiang (2/2) 卢煜坤 (1/1)
include 卢煜坤, yanmingxiang (2/2) 卢煜坤 (1/1)
src/runtime 卢煜坤, yanmingxiang (2/2) 卢煜坤 (1/1)
src/runtime/cmake/runtime.cmake 卢煜坤, yanmingxiang (2/2) 卢煜坤 (1/1)
tests/ut/runtime 卢煜坤, yanmingxiang (2/2) 卢煜坤 (1/1)

💡 Tip:

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

CLA Signature Pass

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

likedislike
zhut66成员
5月29日 评论:

compile

likedislike
此处折叠了430条消息 查看更多
CANN-robotCANN-robot成员
6月10日 关闭了关联的issue
CANN-robotCANN-robot成员
6月10日 合入了pull request
ykl999ykl999成员
6月20日 删除了关联的issue:[Requirement|需求建议]: aclgraph UB 扩流适配
ykl999ykl999成员
6月20日 关联了issue:CANN Runtime Roadmap (2026 Q2)
CANN-robotCANN-robot成员
6月20日 删除了label:ci-pipeline-passed