将 DV121中 EmbeddingBag 算子的相关能力反合至 ops-nn 主线,新增 ascend960dt 平台适配。本次需求包括算子的编译与二进制配置、2D 路径的 GatherDMA 优化模板,以及 arch35/arch92 双入口版本隔离,同时复用主线已有的 tiling、infershape 与 kernel 公共实现。
DV121EmbeddingBag 算子蓝区反合
将算子的编译、注册和二进制配置扩展至 ascend960dt。为 DAV_9201 架构引入 EmbeddingBag 2D 路径的 GatherDMA 优化,使用 GatherDataCopy 按索引批量搬入权重行,替代逐行 DataCopyPad,提升大 bag 场景的搬运效率。通过 arch35/arch92 双入口与公共头复用主线实现,减少架构间的重复代码。隔离 arch92 专属 API,恢复 950 等既有平台的可编译性。对齐 kernel 侧 gatherIndexBuf_ 分配量与 host 侧 UB 预算口径,消除 UB 越界风险。
nan
/assign
Backgroud(背景信息)
将 DV121中 EmbeddingBag 算子的相关能力反合至 ops-nn 主线,新增 ascend960dt 平台适配。本次需求包括算子的编译与二进制配置、2D 路径的 GatherDMA 优化模板,以及 arch35/arch92 双入口版本隔离,同时复用主线已有的 tiling、infershape 与 kernel 公共实现。
Origin(信息来源)
DV121EmbeddingBag 算子蓝区反合
Benefit / Necessity (价值/作用)
将算子的编译、注册和二进制配置扩展至 ascend960dt。为 DAV_9201 架构引入 EmbeddingBag 2D 路径的 GatherDMA 优化,使用 GatherDataCopy 按索引批量搬入权重行,替代逐行 DataCopyPad,提升大 bag 场景的搬运效率。通过 arch35/arch92 双入口与公共头复用主线实现,减少架构间的重复代码。隔离 arch92 专属 API,恢复 950 等既有平台的可编译性。对齐 kernel 侧 gatherIndexBuf_ 分配量与 host 侧 UB 预算口径,消除 UB 越界风险。
Design(设计方案)
nan