c++ - ¿Obtener miembro de__m128 por índice?

clang sse

Question

Tengo un código, originalmente dado por alguien que trabaja con MSVC, y estoy tratando de que funcione en Clang. Aquí está la función con la que estoy teniendo problemas:

float vectorGetByIndex( __m128 V, unsigned int i )
{
assert( i <= 3 );
return V.m128_f32[i];
}

El error que recibo es el siguiente:

Member reference has base type ''__m128'' is not a structure or union.

Miré a mi alrededor y descubrí que Clang (y quizás GCC) tiene un problema con el tratamiento de __m128 como una estructura o unión. Sin embargo, no he logrado encontrar una respuesta directa sobre cómo puedo recuperar estos valores. He intentado usar el operador de subíndices y no pude hacer eso, y he echado un vistazo a la enorme lista de funciones intrínsecas SSE y todavía no he encontrado una adecuada.

Answer 1

Como una modificación de la solución de hirschhornsalz, si i es una constante de tiempo de compilación, podría evitar la ruta de la unión por completo utilizando una aleatoria / tienda:

template<unsigned i>
float vectorGetByIndex( __m128 V)
{
#ifdef __SSE4_1__
return _mm_extract_epi32(V, i);
#else
float ret;
// shuffle V so that the element that you want is moved to the least-
// significant element of the vector (V[0])
V = _mm_shuffle_ps(V, V, _MM_SHUFFLE(i, i, i, i));
// return the value in V[0]
return _mm_cvtss_f32(V);
#endif
}

Answer 2

Incluso si SSE4.1 está disponible y i es una constante de tiempo de compilación, no puede usar pextract etc. de esta manera:

// broken code starts here
template<unsigned i>
float vectorGetByIndex( __m128 V) {
return _mm_extract_epi32(V, i);
}
// broken code ends here

No lo borro porque es un recordatorio útil de cómo no hacer las cosas y dejar que sea una humillación pública.

Mejor uso

template<unsigned i>
float vectorGetByIndex( __m128 V) {
union {
__m128 v;
float a[4];
} converter;
converter.v = V;
return converter.a[i];
}

que funcionará independientemente del conjunto de instrucciones disponibles.

Answer 3

La forma en que uso es

union vec { __m128 sse, float f[4] };
float accessmember(__m128 v, int index)
{
vec v.sse = v;
return v.f[index];
}

Parece que funciona bastante bien para mí.

Answer 4

Una unión es probablemente la forma más portátil de hacer esto:

union {
__m128 v; // SSE 4 x float vector
float a[4]; // scalar array of 4 floats
} U;
float vectorGetByIndex(__m128 V, unsigned int i)
{
U u;
assert(i <= 3);
u.v = V;
return u.a[i];
}