Closed
[Requirement|需求建议]: A5&A6支持strided_slice_assign_v2算子 #4313
liaohumingAdd on 8 days agoClosed at 7 days ago
8 days ago Added label: requirement
8 days ago Assign liaohuming as responsible person
7 days ago Close issue
7 days ago Added label: resolved
一、背景信息 (必填)
当前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 算子约束限制
💡 备注(选填)