利用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