Closed
[Requirement|需求建议]: ChunkCat、StatelessSampleMultinonmial算子支持Ascend960DT #3763
warmingAdd on  20 days agoClosed at  19 days ago
warming
20 days ago Create

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的性能优化

likedislike
陈思陈思Collaborators
19 days ago Assign wsunmoon as responsible person
陈思陈思Collaborators
19 days ago Added label: pending-confirmation
CANN-robotCANN-robotCollaborators
19 days ago Close issue
CANN-robotCANN-robotCollaborators
19 days ago Added label: resolved