Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
需要在ascend950(arch35)/ A2/A3平台新增fused_adamw 算子
来源于 cann/ops-nn 算子仓库 PR #7851(作者 @佐助大王)。
为解决fdsp类型模型,新增优化器optim中adamw算子的fused模式,使输入可输入tensorlist。减少for循环调用输入tensor的adamw算子的调用开销。
/assign
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(设计方案)