已合并
【feat】: 为 MallocReadOnlyDevArgs 添加 v2 动态图支持 #3655
yuht9创建于 6月22日
【feat】: 为 MallocReadOnlyDevArgs 添加 v2 动态图支持 #3655
已合并
yuht9创建于 6月22日
yuht9成员
6月22日

Pull Request

描述

新增 EagerArgsHandler 类,使自定义算子可在 v2 (RT2.0) 动态执行路径中
调用 MallocReadOnlyDevArgs() 分配设备侧只读 args。

核心实现:

  • 新增 EagerArgsHandler 类,实现 ArgsHandler 接口
  • 延迟初始化:OutputsCreator 创建空对象,RunFunc 注入 allocator
  • 管理设备内存分配和释放
  • 析构时自动清理,支持幂等 Release()
  • 重构 AdditionalInputIndex/AdditionalOutputIndex 枚举
  • kArgsHandler 从输入移至输出
  • CustomOpInput 从 AdditionalInputIndex::kNum 开始
  • v2 Lowering 新增 FreeArgsGuarder 节点,自动清理 args 内存
  • v1 CustomTaskInfo 适配:kArgsHandler 移至 outputs
  • 使用 GetValue() 从 Chain 读取指针值

测试:

  • 新增 EagerArgsHandler 单元测试(9 个用例)
  • 新增 v2 集成测试 custom_op_malloc_read_only_dev_args_test
  • 更新 update_args_context_unittest(4 个用例)和 custom_node_converter_unittest

变更类型

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

关联的Issue

如何测试

描述测试此变更的步骤和前提条件:

  1. 动态图场景,MallocReadOnlyDevArgs功能是否正常

核对清单

其他信息

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

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 yuht9 的贡献)
Yyuht9成员
6月22日 添加了label:enhancement
Yyuht9成员
6月22日 创建了 pull request,commit 4386999a
CANN-robotCANN-robot成员
6月22日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
6月22日 评论:

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 yangyongqiang0606, hugo111, Mexyy (3/2) yangyongqiang0606 (1/1)

💡 Tip:

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

CLA Signature Pass

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

likedislike
Yyuht9成员
6月22日 修改了pull request 的描述
此处折叠了313条消息 查看更多
夏国正成员
6月26日 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
6月26日 添加了label:lgtm
yangyongqiang
yangyongqiang成员
6月26日 评论:

/approve

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