Как написать такое при помощи AVX?

Question

[Ketchuuuup]

Как написать такое при помощи AVX?

Если элемент массива не входит в заданный диапазон, то присвоить ему значение 0.

Как бы я написал на C++:

if (item < min || item > max)
{
   item = 0;
}

item, min, max имеют тип std::uint16_t.
Я использую 256-битные векторы (__m256i).

Answer 1

[Wataru]

Каким-нибудь _mm256_cmpgt_epi16 два раза можно получить для каждого числа 0xFFFF, если оно входит в интервал и 0 иначе. Потом каким-нибудь _mm256_and_si256 можно занулить те числа, которые не попали. Возможно, придется делать and 2 раза.

Comments

[Ketchuuuup]

_mm256_cmpgt_epi16 не подходит, у меня же unsigned

[Wataru]

Ну посмотрите, что там еще за функции есть с cmp в названии. Может, какая-то подойдет.

[Ketchuuuup]

Wataru, нет, уже проверял. так же как и _mm256_cmplt_epi16

[Wataru]

Ketchuuuup, ну приведите ваши числа тогда к signed, вычтя 128 из них всех.

[Wataru]

Ketchuuuup, Или 32768, если они у вас 16-битные.