[昇腾CANN自定义算子]TIK算子矢量计算接口vec_add

利用AI Core的Vector Unit,进行矢量加计算。

vec_add(mask, dst, src0, src1, repeat_times, dst_rep_stride, src0_rep_stride, src1_rep_stride)
  • mask
    通常用这个参数的连续模式,就是设置前N个数据参与计算。
    每次迭代可以计算256字节的数据。
    比如,当计算float16类型数据时,每个数据2字节,所以mask设置成128时,就可以在一个迭代中计算256字节的数据。

  • dst, src0, src1
    输出与输入Tensor。
    通常事先通过tik_instance.Tensor定义定义好。

  • repeat_times
    迭代次数。(最大值为255)
    比如当Tensor大小是512字节时,因为一次迭代会计算256字节,所以应该迭代2次。

  • dst_rep_stride, src0_rep_stride, src1_rep_stride
    步长。两次迭代头与头之间的block数。
    通常连续计算时,设置为8。(32B*8=256B)


注:当前版本为CANN 5.0

上一篇:一些铜牌题


下一篇:USACO 1.6.1 数字三角形