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 算子约束限制
💡 备注（选填）