已开启
[需求] 新增 DynamicAUGRU 算子(Ascend 950 / arch35,支持动态图推理) #6086
Huang-Peng创建于  4 天前
Huang-Peng成员
4 天前 创建

背景

DynamicAUGRU(Attention-based Update Gate GRU,由注意力分数调节更新门的门控循环单元)用于处理用户行为等序列数据。ops-nn 仓库当前无该算子实现,需要新增以支持 Ascend 950(arch35)及图模式动态 Shape 推理。

需求

  1. 算子原型与图模式推理:定义 DynamicAUGRU 原型(x [T,B,I] 与权重、注意力分数输入,7 个 [T,B,H] 输出),实现图模式推理与动态 Shape 推导。
  2. Host 侧实现:形状推导(InferShape)、通用 tiling、arch35 tiling,并在 op_host 注册算子(Ascend 950)。
  3. Kernel 实现:arch35 内核主体,覆盖 fp16 输入与 fp16/fp32 输出场景。
  4. 序列控制语义:seq_length 支持 INT32 [B](有效序列长度)与 FLOAT16 [T,B,H](逐元素掩码)两种模式;bias_input/bias_hidden/init_h 为可选输入,缺省按零处理。
  5. 测试与文档:补充 Host UT(InferShape/tiling)、图模式 UT、CPU golden 与输入生成、GE 图样例;新增算子 README 并在 docs/zh/op_list.md 注册。

验收标准

  • 算子可在 Ascend 950 上构建、安装
  • Host UT 与图模式 UT 通过
  • kernel 精度与 CPU golden 对齐
  • 动态 Shape 场景经 GE 图样例验证
likedislike
yuning_chenyuning_chen成员
4 天前 将 Huang-Peng 设为负责人