Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
StackBallQuery 增加950支持
geir、kernel
全数据类型
新增infershape和tiling拦截。
新增模板参数,修复kernel不支持int64 batchcount的问题
950
💡 备注(选填)
/assign @wang-shilong32
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
一、背景信息 (必填)
StackBallQuery 增加950支持
二、价值/作用 (必填)
StackBallQuery 增加950支持
三、设计方案 (必填)
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
geir、kernel
3.2 总体设计
3.2.1 算子支持的数据类型
全数据类型
3.2.2 host侧设计
新增infershape和tiling拦截。
3.2.3 kernel侧设计
新增模板参数,修复kernel不支持int64 batchcount的问题
3.3 支持硬件
950
3.4 算子约束限制
💡 备注(选填)