Как я могу взять обратное (обратное) число с плавающей запятой с инструкциями SSE, но только для ненулевых значений?
Фон ниже:
Я хочу нормализовать массив векторов, чтобы каждое измерение имело одинаковое среднее значение. В C это можно закодировать как:
float vectors[num * dim]; // input data
// step 1. compute the sum on each dimension
float norm[dim];
memset(norm, 0, dim * sizeof(float));
for(int i = 0; i < num; i++) for(int j = 0; j < dims; j++)
norm[j] += vectors[i * dims + j];
// step 2. convert sums to reciprocal of average
for(int j = 0; j < dims; j++) if(norm[j]) norm[j] = float(num) / norm[j];
// step 3. normalize the data
for(int i = 0; i < num; i++) for(int j = 0; j < dims; j++)
vectors[i * dims + j] *= norm[j];
Теперь из соображений производительности я хочу сделать это с помощью SSE intinsics. Setp 1 и шаг 3 просты, но я застрял на шаге 2. Кажется, я не нашел ни одного примера кода или очевидной инструкции SSE, чтобы получить обратное значение if оно не равно нулю . Для деления _mm_rcp_ps делает свое дело и, возможно, комбинирует его с условным перемещением, но как получить маску, указывающую, какой компонент равен нулю?
Мне не нужен код к алгоритму, описанному выше, просто функция "обратная, если не нулевая":
__m128 rcp_nz_ps(__m128 input) {
// ????
}
Спасибо!