Closed
[Requirement|需求建议]: A5&A6支持strided_slice_assign_v2算子 #4313
liaohumingAdd on  8 days agoClosed at  7 days ago
liaohuming
liaohumingCollaborators
8 days ago Create

一、背景信息 (必填)

当前A5A6不支持strided_slice_assign_v2,为满足需要,进行适配。

二、价值/作用 (必填)

A5A6支持strided_slice_assign_v2算子。

三、设计方案 (必填)

3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
3.2 总体设计
3.2.1 算子支持的数据类型

FLOAT16、FLOAT、BFLOAT16、INT32、INT64、DOUBLE、INT8七种数据类型,与A3保持一致。

3.2.2 host侧设计

根据shape特点,为每个模板进行不一样的切分。

3.2.3 kernel侧设计

设计如下三个模板:
(1)模板1:SIMD,SIMT混合编程,SIMD连续搬入,SIMT离散写出。
(2)模板2:DMA直通,纯搬运模板,将input_value一行一行搬入,然后搬出。
(3)模板3:Scatter模板,将input_value与var同时搬入,将input_value 散列到var中的目标区域,然后将var搬出。

3.3 支持硬件

3.4 算子约束限制

💡 备注(选填)

likedislike
liaohumingliaohumingCollaborators
8 days ago Added label: requirement
liaohumingliaohumingCollaborators
8 days ago Assign liaohuming as responsible person
CANN-robotCANN-robotCollaborators
7 days ago Close issue
CANN-robotCANN-robotCollaborators
7 days ago Added label: resolved