Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
提升stateless_random_uniform和stateless_random_normal两个算子的性能
性能提升
总体思路:收编小算子,将原有的在aclnn侧的多个小算子拼接实现功能归一至kernel侧,调用一个算子即可
💡 备注(选填)
/assign @gh_M
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
一、背景信息 (必填)
提升stateless_random_uniform和stateless_random_normal两个算子的性能
二、价值/作用 (必填)
性能提升
三、设计方案 (必填)
总体思路:收编小算子,将原有的在aclnn侧的多个小算子拼接实现功能归一至kernel侧,调用一个算子即可
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
3.2 总体设计
3.2.1 算子支持的数据类型
3.2.2 host侧设计
3.2.3 kernel侧设计
3.3 支持硬件
3.4 算子约束限制
💡 备注(选填)