有条件的 SSE/AVX 根据比较添加或归零元素

Question

我有以下 __m128 vector :

v_weight

v_entropy

只有当 v_weight 中的元素不为 0f 时，我才需要将 v_entropy 添加到 v_weight。

显然 _mm_add_ps() 添加了所有元素。

我可以编译成 AVX，但不能编译成 AVX2。

编辑

我事先知道 v_weight 中有多少元素将为 0(总是有 0 或最后 1、2 或 3 个元素)。如果更简单，我如何将 v_entropy 中的相应元素清零？

Answer 1

cmpeq/cmpgt 指令创建一个掩码，全 1 或全 0。整体流程如下:

auto mask=_mm_cmpeq_ps(_mm_setzero_ps(), w);
mask=_mm_andnot_ps(mask, entropy);
w = _mm_add_ps(w, mask);

其他选项是无论如何都要积累，但使用 blendv 在添加/不添加之间进行选择。

auto w2=_mm_add_ps(e,w);
auto mask=_mm_cmpeq_ps(zero,w);
w=_mm_blendv_ps(w2,w, mask);

第三个选项使用 w+e = 0 的事实，当 w=0 时

 m=(w==0); // make mask as in above
 w+=e; // add
 w&=~m; // revert adding for w==0

(我使用 cmpeq 而不是 cmpneq 以使其也可用于整数。)