我可以用simd分类吗?

我对simd很好奇,想知道它是否能处理这个用例。

假设我有一个2048个整数的数组,比如 [0x018A、0x004B、0x01C0、0x0234、0x098、0x0343、0x0222、0x0301、0x0398、0x087、0x0167、0x0389、0x03F2、0x0034、0x0345…]

请注意它们都是如何以0x00、0x01、0x02或0x03开头的。我想把它们分成4个数组:

• 一个用于所有以0x00开头的整数
• 一个用于所有以0x01开头的整数
• 一个用于所有以0x02开头的整数
• 一个用于所有以0x03开头的整数

我想我会有这样的代码:

int main() {
   uint16_t in[2048] = ...;

   // 4 arrays, one for each category
   uint16_t out[4][2048];

   // Pointers to the next available slot in each of the arrays
   uint16_t *nextOut[4] = { out[0], out[1], out[2], out[3] };

   for (uint16_t *nextIn = in; nextIn < 2048; nextIn += 4) {

       (*** magic simd instructions here ***)

       // Equivalent non-simd code:
       uint16_t categories[4];
       for (int i = 0; i < 4; i++) {
           categories[i] = nextIn[i] & 0xFF00;
       }
       for (int i = 0; i < 4; i++) {
           uint16_t category = categories[i];
           *nextOut[category] = nextIn[i];
           nextOut[category]++;
       }
   }
   // Now I have my categoried arrays!
}

我想我的第一个内环不需要SIMD,它可能只是一个 (x & 0xFF00FF00FF00FF00) 指令,但我想知道我们是否可以把第二个内部循环变成一个SIMD指令。

对于我正在做的这个“分类”操作,是否有任何类型的SIMD指令?

“插入”说明似乎有点好用,但我太过环保,无法理解 https://software.intel.com/en-us/node/695331 .

如果没有,有什么接近吗?

谢谢!