计算 sse var 的最大掩码的最佳方法

Question

（我只对前 3 个组件感兴趣）

例如：
[ 1 2 3 ? ] 应该产生[ 0 0 -1 ? ]

此外，只设置一个“位”很重要，这样：
[ 1 2 2 ? ] 不应产生 [ 0 -1 -1 ? ]
而是
[ 0 -1 0 ? ] 或 [ 0 0 -1 ? ]（不管是哪一个）

后面的（坏的）解决方案是可能的，例如通过提取水平最大值并与原始解决方案进行比较：

__m128 abcd; // input
__m128 ccac           = _mm_shuffle_ps(abcd, abcd, 0x8A);
__m128 abcd_ccac      = _mm_max_ps(abcd, ccac);
__m128 babb           = _mm_shuffle_ps(abcd, abcd, 0x51);
__m128 abcd_ccac_babb = _mm_max_ps(abcd_ccac, babb);
__m128 mask           = _mm_cmpeq_ps(abcd, abcd_ccac_babb);

也许一些按位运算来消除重复的设置位？

更新：

跟进，我已经提出了另一个（糟糕的）解决方案。
关键是将每个组件与另一个组件进行比较，避免等式（在其他地方没有 a >= b 和 b >= a）。

a > b & a >= c
b > c & b >= a
c > a & c >= b

屈服：

([ a b c ? ] > [ b c a ? ]) & ([ a b c ? ] >= [ c a b ? ])

在代码中：

__m128 abcd; // input
__m128 bcad = _mm_shuffle_ps(abcd, abcd, 0xC9);
__m128 gt   = _mm_cmpgt_ps(abcd, bcad);
__m128 cabd = _mm_shuffle_ps(abcd, abcd, 0xD2);
__m128 ge   = _mm_cmpge_ps(abcd, cabd);
__m128 mask = _mm_and_ps(gt, ge);

在[ x x x ? ] 的情况下失败（产生[ 0 0 0 ? ]）。

接近:-)

有什么想法吗？

更新：

我现在正在使用以下解决方案：

__m128  abcd; // input
__m128  bcad = _mm_shuffle_ps(abcd, abcd, 0xC9);
__m128  gt   = _mm_cmpgt_ps(abcd, bcad);
__m128  cabd = _mm_shuffle_ps(abcd, abcd, 0xD2);
__m128  ge   = _mm_cmpge_ps(abcd, cabd);
__m128  and  = _mm_and_ps(gt, ge);
__m128i bits = _mm_setr_epi32(_mm_movemask_ps(and), -1, -1, -1);
__m128i dirt = _mm_cmpeq_epi32(bits, _mm_setzero_si128());
__m128i mask = _mm_or_si128(dirt, _mm_castps_si128(and));

Answer 1

我没有对此进行测试，但我相信它只会在最大值的第一个（最高阶）出现中为您提供 -1：

__m128 abcd; // input
__m128 ccac = _mm_shuffle_ps( abcd,abcd,0x8A );
__m128 abcd_ccac = _mm_max_ps( abcd,ccac );
__m128 babb = _mm_shuffle_ps( abcd,abcd,0x51 );
__m128 abcd_ccac_babb = _mm_max_ps( abcd_ccac,babb );
__m128 mask = _mm_cmpeq_ps( abcd,abcd_ccac_babb );

// set the '?' position in mask to zero
mask = _mm_blend_ps( mask,_mm_setzero_ps(),0x08 );
// shift mask left 32 bits shifting in zeros
__m128 maskSrl32 = _mm_shuffle_ps( mask,mask,_MM_SHUFFLE( 3,1,0,3 ) );
// shift mask left 64 bits shifting in zeros
__m128 maskSrl64 = _mm_shuffle_ps( mask,mask,_MM_SHUFFLE( 3,0,3,3 ) );
// andnot the shifted masks with mask
// in doing so, the higher order set bits will suppress any set bits which follow
mask = _mm_andnot_ps( maskSrl32,mask );
mask = _mm_andnot_ps( maskSrl64,mask );
// select -1 using the final mask
__m128 result = _mm_and_ps( mask,_mm_set1_ps( -1.0f ) );

反转移动方向以在最低阶最大位置产生 -1。