Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
打通推荐模型功能断点
涉及aclnn及pytorch
fp16/bf16/fp32
使用Broadcast模板
依据公式泰勒展开实现
Ascend 950PR/Ascend 950DT
💡 备注(选填)
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
一、背景信息 (必填)
打通推荐模型功能断点
二、价值/作用 (必填)
打通推荐模型功能断点
三、设计方案 (必填)
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
涉及aclnn及pytorch
3.2 总体设计
3.2.1 算子支持的数据类型
fp16/bf16/fp32
3.2.2 host侧设计
使用Broadcast模板
3.2.3 kernel侧设计
依据公式泰勒展开实现
3.3 支持硬件
Ascend 950PR/Ascend 950DT
3.4 算子约束限制
💡 备注(选填)