Особенности SSE3: Как найти максимум большого массива с плавающей точкой

У меня есть следующий код, чтобы найти максимальное значение

int length = 2000;
float *data;
// data is allocated and initialized

float max = 0.0;
for(int i = 0; i < length; i++)
{
if(data[i] > max)
{
max = data;
}
}

Я пытался векторизовать его с помощью встроенных функций SSE3, но я немного удивлен тем, как мне следует проводить сравнение.

int length = 2000;
float *data;
// data is allocated and initialized

float max = 0.0;
// for time being just assume that length is always mod 4
for(int i = 0; i < length; i+=4)
{
__m128 a = _mm_loadu_ps(data[i]);
__m128 b = _mm_load1_ps(max);

__m128 gt = _mm_cmpgt_ps(a,b);

// Kinda of struck on what to do next
}

Может кто-нибудь дать некоторое представление об этом.

7

Решение

Таким образом, ваш код находит наибольшее значение в массиве фиксированной длины с плавающей точкой. ХОРОШО.

Существует _mm_max_ps, который дает вам попарные максимумы из двух векторов по четыре числа с плавающей запятой в каждом. Так как насчет этого?

int length = 2000;
float *data; // maybe you should just use the SSE type here to avoid copying later
// data is allocated and initialized

// for time being just assume that length is always mod 4
__m128 max = _mm_loadu_ps(data); // load the first 4
for(int i = 4; i < length; i+=4)
{
__m128 cur = _mm_loadu_ps(data + i);
max = _mm_max_ps(max, cur);
}

Наконец, возьмите самое большое из четырех значений в max (увидеть Получить максимальное значение в векторе __m128i с SSE? для этого).

Это должно работать так:

Шаг 1:

[43, 29, 58, 94] (this is max)
[82, 83, 10, 88]
[19, 39, 85, 77]

Шаг 2:

[82, 83, 58, 94] (this is max)
[19, 39, 85, 77]

Шаг 2:

[82, 83, 85, 94] (this is max)
10

Другие решения

Других решений пока нет …

По вопросам рекламы [email protected]