ARM NEON: как реализовать логику типа mm_mask_compress? ⇐ C++

[Anonymous]

Функция с именем filter_range предназначена для сохранения элементов nums, для которых filter != 0, и удаления всех остальных. Его логика следующая.
static size_t filter_range(int* nums, const uint8_t* filter, size_t size) {
size_t result_offset = 0;
for (auto i = 0; i < size; i++) {
if (filter) {
*(nums + result_offset) = *(nums + i);
result_offset++;
}
}
return result_offset;
}

Для AVX512 от AMD можно удобно использовать инструкцию _mm512_mask_compress_epi для достижения этой функциональности. Можно ли в ARM NEON использовать инструкции SIMD для максимального ускорения этой функции?

Вдохновленные этой статьей, мы смогли получить совпадения, каждые четыре бита которого представляют значение фильтра. Тогда мы могли бы использовать совпадения &= (matches - 1) и __builtin_ctzll для перебора битов, что называется алгоритмом Кернигана.
Однако это все равно хуже, чем версия AVX512, если использовать _mm512_mask_compress_epi.

static size_t filter_range2(int* data, const uint8_t* filter, size_t size) {
constexpr size_t data_type_size = sizeof(int);
size_t start_offset = 0;
size_t result_offset = 0;

#if defined(__ARM_NEON__) || defined(__aarch64__)
std::cout > 2;
*(data + result_offset++) = *(data + index);
}
}

start_offset += kBatchNums;
filter += kBatchNums;
}
#endif

for (auto i = start_offset; i < size; ++i) {
if (filter) {
*(data + result_offset) = *(data + i);
result_offset++;
}
}

return result_offset;
}


Подробнее здесь: https://stackoverflow.com/questions/783 ... k-compress