Closed
[Requirement|需求建议]: ChunkCat、StatelessSampleMultinonmial算子支持Ascend960DT #3763
warmingAdd on 20 days agoClosed at 19 days ago
19 days ago Assign wsunmoon as responsible person
19 days ago Added label: pending-confirmation
19 days ago Close issue
19 days ago Added label: resolved
Backgroud(背景信息)
ChunkCat, StatelessSampleMultinonmial算子当前仅在 ascend950 等上提供支持。本需求为ChunkCat, StatelessSampleMultinonmial算子新增 Ascend960DT平台支持,使其可在Ascend960DT上完成算子注册、编译并运行。
Origin(信息来源)
自提
Benefit / Necessity (价值/作用)
扩展 ChunkCat, StatelessSampleMultinonmial算子的产品支持范围,使 Ascend960DT平台可使用ChunkCat, StatelessSampleMultinonmial算子计算能力,满足相关网络结构在该平台上的算子需求。
Design(设计方案)
aclnn接口适配Ascend960DT
CMakeLists.txt添加Ascend960DT
算子信息信息库新增支持Ascend960DT
新增Ascend960DT二进制配置文件
ChunkCat算子在960DT上做先切多核再切UB的性能优化