昇腾社区首页
中文
注册

vsel

功能说明

从两个输入中根据select mode选择元素输出。

函数原型

void vsel(__ubuf__ half *dst, __ubuf__ half *src0, __ubuf__ void *src1, uint8_t repeat, uint8_t dstBlockStride, uint8_t src0BlockStride, uint8_t src1BlockStride, uint8_t dstRepeatStride, uint8_t src0RepeatStride, uint8_t src1RepeatStride, uint8_t selectMode); 
 
void vsel(__ubuf__ float *dst, __ubuf__ float *src0, __ubuf__ float *src1, uint8_t repeat, uint8_t dstBlockStride, uint8_t src0BlockStride, uint8_t src1BlockStride, uint8_t dstRepeatStride, uint8_t src0RepeatStride, uint8_t src1RepeatStride, uint8_t selectMode);

参数说明

表1 vsel特有参数说明

参数名

说明

selectMode

选择模式:

  • 0:根据get_cmpmask信息进行选取
  • 1:选择src0中的元素
  • 2:选择src1中的元素

流水类型

PIPE_V