Segsum算子在Ascend910B上是对应功能实现,但是在Ascend950上没有,需要对标功能完成其在Ascend950上支持
网络支持
功能完全对标Ascend910B算法是实现,性能上对标竞品A100
aclnn通路和geir通路
支持除double之前的其他浮点型
按行进行多核切分
membase方式实现
💡 备注(选填)
一、背景信息 (必填)
Segsum算子在Ascend910B上是对应功能实现,但是在Ascend950上没有,需要对标功能完成其在Ascend950上支持
二、价值/作用 (必填)
网络支持
三、设计方案 (必填)
功能完全对标Ascend910B算法是实现,性能上对标竞品A100
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
aclnn通路和geir通路
3.2 总体设计
3.2.1 算子支持的数据类型
支持除double之前的其他浮点型
3.2.2 host侧设计
按行进行多核切分
3.2.3 kernel侧设计
membase方式实现
3.3 支持硬件
3.4 算子约束限制
💡 备注(选填)