Функция с именем 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
