已关闭
[Requirement|需求建议]: TensorEqual & RandomUniformV2新增ascend960平台支持 #3607
Wangdongxu_mj007创建于 14 天前关闭于 10 天前
14 天前 添加了label:requirement
Wangdongxu_mj007
14 天前 评论:
14 天前 评论:
/assign


14 天前 将 magicjason0007 设为负责人
10 天前 关闭了 issue
10 天前 添加了label:resolved
一、背景信息 (必填)
ascend960 为新增芯片形态。ops-math 仓库中TensorEqual算子和RandomUniformV2算子目前仅注册了 ascend950 / ascend350 等平台,缺少 ascend960 支持,需补齐平台注册、编译单元与二进制配置。
二、价值/作用 (必填)
完成 ascend960 适配后,相关算子可在新平台上被算子信息库发现并正确选择 kernel 二进制,保障新形态芯片的算子可用性。
三、设计方案 (必填)
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
aclnn 直调;本需求涉及的算子编译类型为 aclnn_exclude。
3.2 总体设计
3.2.1 算子支持的数据类型
沿用各算子现有支持的数据类型;为 ascend960 新增对应的_binary.json 二进制配置。
3.2.2 host侧设计
tiling侧部分改动采取编译宏隔离。
3.2.3 kernel侧设计
无 kernel 代码逻辑改动,复用现有 arch35 kernel。
3.3 支持硬件
ascend960
3.4 算子约束限制
本需求仅补齐 ascend960 平台适配,不改变算子既有功能与约束,不涉及新增数据类型或泛化能力。
💡 备注(选填)