Closed
[Requirement|需求建议]: 【split_tensor】【910B、910_93】int16、int32、bool数据类型路由到AICORE上运行 #3673
tangkaidiAdd on 22 days agoClosed at 22 days ago
22 days ago Change title to "[Requirement|需求建议]: 【split_tensor】【910B、910_93】int16、int32、bool数据类型路由到AICORE上运行", originally titled "[Requirement|需求建议]: 【split_tensor】int16、int32、bool数据类型路由到AICORE上运行".
陈思
22 days ago Comments:
22 days ago Comments:
/assign @tangkaidi


22 days ago Assign tangkaidi as responsible person
22 days ago Close issue
22 days ago Added label: resolved
一、背景信息 (必填)
调用aclnnSplitTensor算子,发现int32数据类型性能不佳,原因是计算走入到了AICPU实现。
需要使能split_tensor算子int16、int32、bool数据类型到AICORE上运行。
二、价值/作用 (必填)
三、设计方案 (必填)
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
3.2 总体设计
aclnn接口L0 AICORE支持数据类型白名单添加int16、int32、bool
3.2.1 算子支持的数据类型
int16、int32、bool
3.2.2 host侧设计
3.2.3 kernel侧设计
3.3 支持硬件
910B、910_93
3.4 算子约束限制
💡 备注(选填)