已关闭
[Requirement|需求建议]: 新增算子fused_adamw #4847
佐助大王创建于  23 天前关闭于  22 天前
佐助大王
佐助大王成员
23 天前 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

Backgroud(背景信息)

需要在ascend950(arch35)/ A2/A3平台新增fused_adamw 算子

Origin(信息来源)

来源于 cann/ops-nn 算子仓库 PR #7851(作者 @佐助大王)。

Benefit / Necessity (价值/作用)

为解决fdsp类型模型,新增优化器optim中adamw算子的fused模式,使输入可输入tensorlist。减少for循环调用输入tensor的adamw算子的调用开销。

Design(设计方案)

likedislike
佐助大王佐助大王成员
23 天前 添加了label:requirement
佐助大王
佐助大王成员
23 天前 评论:

/assign

likedislike
CANN-robotCANN-robot成员
23 天前 将 ji-songyuan 设为负责人
CANN-robotCANN-robot成员
22 天前 关闭了 issue
CANN-robotCANN-robot成员
22 天前 添加了label:resolved
佐助大王佐助大王成员
22 天前 关联了pull request:docs_change