vec_or

功能说明

每对elements按位或运算：

函数原型

vec_or(mask, dst, src0, src1, repeat_times, dst_rep_stride, src0_rep_stride, src1_rep_stride)

参数说明

请参见参数说明。

dst/src0/src1的数据类型需要保持一致。

Atlas 200/300/500 推理产品，dst/src0/src1支持的数据类型为：Tensor(uint16/int16)

Atlas 训练系列产品，dst/src0/src1支持的数据类型为：Tensor(uint16/int16)

Atlas推理系列产品AI Core，dst/src0/src1支持的数据类型为：Tensor(uint16/int16)

Atlas推理系列产品Vector Core，dst/src0/src1支持的数据类型为：Tensor(uint16/int16)

Atlas A2训练系列产品/Atlas 800I A2推理产品，dst/src0/src1支持的数据类型为：Tensor(uint16/int16)

Atlas 200/500 A2推理产品，dst/src0/src1支持的数据类型为：Tensor(uint8/int8/uint16/int16)

返回值

无

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas推理系列产品AI Core

Atlas推理系列产品Vector Core

Atlas A2训练系列产品/Atlas 800I A2推理产品

Atlas 200/500 A2推理产品

注意事项

其他请参考注意事项。

调用示例

此样例是针对数据量较小、一次搬运就可以完成的场景，目的是让大家了解接口的功能，更复杂的数据量较大的样例可参见调用示例

from tbe import tik
tik_instance = tik.Tik()
src0_gm = tik_instance.Tensor("int16", (128,), name="src0_gm", scope=tik.scope_gm)
src1_gm = tik_instance.Tensor("int16", (128,), name="src1_gm", scope=tik.scope_gm)
dst_gm = tik_instance.Tensor("int16", (128,), name="dst_gm", scope=tik.scope_gm)
src0_ub = tik_instance.Tensor("int16", (128,), name="src0_ub", scope=tik.scope_ubuf)
src1_ub = tik_instance.Tensor("int16", (128,), name="src1_ub", scope=tik.scope_ubuf)
dst_ub = tik_instance.Tensor("int16", (128,), name="dst_ub", scope=tik.scope_ubuf)
# 将用户输入数据从gm搬运到ub
tik_instance.data_move(src0_ub, src0_gm, 0, 1, 8, 0, 0)
tik_instance.data_move(src1_ub, src1_gm, 0, 1, 8, 0, 0)
tik_instance.vec_or(128, dst_ub, src0_ub, src1_ub, 1, 8, 8, 8)
# 将计算结果从ub搬运到目标gm
tik_instance.data_move(dst_gm, dst_ub, 0, 1, 8, 0, 0)

tik_instance.BuildCCE(kernel_name="vec_or", inputs=[src0_gm, src1_gm], outputs=[dst_gm])

结果示例：

输入数据(src0_gm):
[-6  5 -8  9  5  7  0  0  5  9 -8 -7  4  3 -7  1  4 -8 -9  8  2  8  0  0
  0  6  8  4  2  7 -9 -1  8 -1  7 -2  1 -1  8  0  1 -7  8  8 -8 -5  8  5
 -4  1 -3  6  6  7 -9 -8  6 -2  9 -2 -6 -1  1 -9 -3 -2  3 -7 -1  5 -9  6
  5  0  0  8  5 -9 -6 -8  5  5  5 -3 -8  9  5  5 -1  0 -6  6  1 -5  0  7
  3  4 -4  8  9  0 -7 -5  5 -1  5 -6 -2  1  8  7 -5 -9  0  3  4 -4 -7  2
 -6  0  3  5  1 -6 -2 -5]

输入数据(src1_gm):
[-8  9 -4  3  3  2  3 -6  9  9 -5  1  3  0 -5  2  8  9 -9  1  5  2  4  1
  7 -5 -3  8 -3  8 -5 -6 -9  7  4 -7 -1 -5  9  2 -3 -3  0  9  1  7 -6 -1
  8 -2  6 -5  5  9  5 -4  9  5  9  4  7  9 -2  9  0 -3 -2  8  4  3 -7 -4
  8  9  5 -4  5 -1  0  1 -2 -3 -1 -7 -1  3  4  8 -9 -3 -6  0  0 -8  2 -7
  7  1 -3 -9 -6  2 -6  7  8 -2 -8  2  4 -2  2 -6  0  0  2  1  3 -9 -8  0
  9 -1  5  2 -8 -5  0  2]

输出数据(dst_gm):
[-6 13 -4 11  7  7  3 -6 13  9 -5 -7  7  3 -5  3 12 -7 -9  9  7 10  4  1
  7 -1 -3 12 -1 15 -1 -1 -1 -1  7 -1 -1 -1  9  2 -3 -3  8  9 -7 -1 -6 -1
 -4 -1 -1 -1  7 15 -9 -4 15 -1  9 -2 -1 -1 -1 -1 -3 -1 -1 -7 -1  7 -1 -2
 13  9  5 -4  5 -1 -6 -7 -1 -3 -1 -3 -1 11  5 13 -1 -3 -6  6  1 -5  2 -1
  7  5 -3 -1 -5  2 -5 -1 13 -1 -3 -6 -2 -1 10 -1 -5 -9  2  3  7 -1 -7  2
 -5 -1  7  7 -7 -5 -2 -5]