Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
针对小尾轴场景的shape，将simt模板进行方案重构以优化性能： 使用segment_ids分核，利用simd计算核内唯一索引的个数和位置，simt对每个唯一索引对应的所有行进行累加。 解决了小尾轴场景性能差的问题。
NA
小尾轴场景shape性能平均提升6倍
/assign
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
Backgroud（背景信息）
针对小尾轴场景的shape，将simt模板进行方案重构以优化性能：
使用segment_ids分核，利用simd计算核内唯一索引的个数和位置，simt对每个唯一索引对应的所有行进行累加。
解决了小尾轴场景性能差的问题。
Origin（信息来源）
NA
Benefit / Necessity （价值/作用）
小尾轴场景shape性能平均提升6倍
Design（设计方案）