2012-05-08 11 views

Respuesta

23

Si necesita firmaron 32x32 bits multiplicación entera entonces el siguiente ejemplo en software.intel.com parece que debería hacer lo que quiera:

static inline __m128i muly(const __m128i &a, const __m128i &b) 
{ 
    __m128i tmp1 = _mm_mul_epu32(a,b); /* mul 2,0*/ 
    __m128i tmp2 = _mm_mul_epu32(_mm_srli_si128(a,4), _mm_srli_si128(b,4)); /* mul 3,1 */ 
    return _mm_unpacklo_epi32(_mm_shuffle_epi32(tmp1, _MM_SHUFFLE (0,0,2,0)), _mm_shuffle_epi32(tmp2, _MM_SHUFFLE (0,0,2,0))); /* shuffle results to [63..0] and pack */ 
} 

Es posible que desee tener dos compilaciones - uno para CPUs viejas y una para CPU recientes, en cuyo caso podría hacer lo siguiente:

static inline __m128i muly(const __m128i &a, const __m128i &b) 
{ 
#ifdef __SSE4_1__ // modern CPU - use SSE 4.1 
    return _mm_mullo_epi32(a, b); 
#else    // old CPU - use SSE 2 
    __m128i tmp1 = _mm_mul_epu32(a,b); /* mul 2,0*/ 
    __m128i tmp2 = _mm_mul_epu32(_mm_srli_si128(a,4), _mm_srli_si128(b,4)); /* mul 3,1 */ 
    return _mm_unpacklo_epi32(_mm_shuffle_epi32(tmp1, _MM_SHUFFLE (0,0,2,0)), _mm_shuffle_epi32(tmp2, _MM_SHUFFLE (0,0,2,0))); /* shuffle results to [63..0] and pack */ 
#endif 
} 
+3

Buena respuesta. Es curioso que hayas hecho exactamente el mismo error tipográfico que tuve una vez en mi código: debería ser _____SSE4_1_____ (sin guiones bajos entre E y 4). Molesto, porque no te das cuenta fácilmente - el programa se ejecuta perfecto siempre que la ruta del código alternativo sea correcta – hirschhornsalz

+1

@drhirsch: gracias por arreglar eso - en código real tiendo a usar '__MNI__',' __SNI__', etc. - principalmente por razones históricas, pero también es menos propenso a errores simples como el anterior. –

+1

¡Impresionante, gracias! Ahora, si hubiera un truco similar para reemplazar '_mm_insert_epi32' en una CPU con SSE2 solamente ... –

7

PMULLD, de SSE 4.1, hace eso.

La descripción es un poco engañosa, habla sobre la multiplicación con signo, pero como solo almacena los 32 bits inferiores, es realmente una instrucción de signo indiferente que puede usarse para ambos, al igual que IMUL.

+0

Thanks. Pero, ¿hay alguna manera de usar solo las instrucciones de SSE 2? – Yury

+2

'_mm_mullo_epi32' si prefiere utilizar elementos intrínsecos que el conjunto sin procesar –

+1

@Leviathan Sí, pero necesita varias instrucciones. Dependiendo de la arquitectura, cuatro 'imul' son posiblemente más rápidos y simples – hirschhornsalz

Cuestiones relacionadas