comparison src/madplug/SFMT-sse2.h @ 1386:b8dd67ad7b86

update SFMT files to version 1.3. please let me know if it break on altivec box.
author Yoshiki Yazawa <yaz@cc.rim.or.jp>
date Fri, 27 Jul 2007 18:56:43 +0900
parents 7e14701aef54
children
comparison
equal deleted inserted replaced
1385:f3c57bbb54d0 1386:b8dd67ad7b86
1 /** 1 /**
2 * @file SFMT-sse2.h 2 * @file SFMT-sse2.h
3 * 3 * @brief SIMD oriented Fast Mersenne Twister(SFMT) for Intel SSE2
4 * @brief SIMD oriented Fast Mersenne Twister(SFMT)
5 * pseudorandom number generator
6 * 4 *
7 * @author Mutsuo Saito (Hiroshima University) 5 * @author Mutsuo Saito (Hiroshima University)
8 * @author Makoto Matsumoto (Hiroshima University) 6 * @author Makoto Matsumoto (Hiroshima University)
9 * 7 *
10 * Copyright (C) 2007 Mutsuo Saito, Makoto Matsumoto and Hiroshima 8 * @note We assume LITTLE ENDIAN in this file
9 *
10 * Copyright (C) 2006, 2007 Mutsuo Saito, Makoto Matsumoto and Hiroshima
11 * University. All rights reserved. 11 * University. All rights reserved.
12 * 12 *
13 * The new BSD License is applied to this software. 13 * The new BSD License is applied to this software, see LICENSE.txt
14 * see LICENSE.txt
15 */ 14 */
16 15
17 #ifndef SFMT_SSE2_H 16 #ifndef SFMT_SSE2_H
18 #define SFMT_SSE2_H 17 #define SFMT_SSE2_H
19 #include <emmintrin.h>
20 18
21 union W128_T { 19 inline static __m128i mm_recursion(__m128i *a, __m128i *b, __m128i c,
22 __m128i si; 20 __m128i d, __m128i mask) ALWAYSINLINE;
23 uint32_t u[4];
24 };
25 21
26 typedef union W128_T w128_t; 22 /**
23 * This function represents the recursion formula.
24 * @param a a 128-bit part of the interal state array
25 * @param b a 128-bit part of the interal state array
26 * @param c a 128-bit part of the interal state array
27 * @param d a 128-bit part of the interal state array
28 * @param mask 128-bit mask
29 * @return output
30 */
31 inline static __m128i mm_recursion(__m128i *a, __m128i *b,
32 __m128i c, __m128i d, __m128i mask) {
33 __m128i v, x, y, z;
34
35 x = _mm_load_si128(a);
36 y = _mm_srli_epi32(*b, SR1);
37 z = _mm_srli_si128(c, SR2);
38 v = _mm_slli_epi32(d, SL1);
39 z = _mm_xor_si128(z, x);
40 z = _mm_xor_si128(z, v);
41 x = _mm_slli_si128(x, SL2);
42 y = _mm_and_si128(y, mask);
43 z = _mm_xor_si128(z, x);
44 z = _mm_xor_si128(z, y);
45 return z;
46 }
27 47
28 #ifdef __GNUC__ 48 /**
29 inline static __m128i mm_recursion(__m128i *a, __m128i *b, __m128i c, 49 * This function fills the internal state array with pseudorandom
30 __m128i d, __m128i mask) 50 * integers.
31 __attribute__((always_inline)); 51 */
32 #else 52 inline static void gen_rand_all(void) {
33 inline static __m128i mm_recursion(__m128i *a, __m128i *b, 53 int i;
34 __m128i c, __m128i d, __m128i mask); 54 __m128i r, r1, r2, mask;
55 mask = _mm_set_epi32(MSK4, MSK3, MSK2, MSK1);
56
57 r1 = _mm_load_si128(&sfmt[N - 2].si);
58 r2 = _mm_load_si128(&sfmt[N - 1].si);
59 for (i = 0; i < N - POS1; i++) {
60 r = mm_recursion(&sfmt[i].si, &sfmt[i + POS1].si, r1, r2, mask);
61 _mm_store_si128(&sfmt[i].si, r);
62 r1 = r2;
63 r2 = r;
64 }
65 for (; i < N; i++) {
66 r = mm_recursion(&sfmt[i].si, &sfmt[i + POS1 - N].si, r1, r2, mask);
67 _mm_store_si128(&sfmt[i].si, r);
68 r1 = r2;
69 r2 = r;
70 }
71 }
72
73 /**
74 * This function fills the user-specified array with pseudorandom
75 * integers.
76 *
77 * @param array an 128-bit array to be filled by pseudorandom numbers.
78 * @param size number of 128-bit pesudorandom numbers to be generated.
79 */
80 inline static void gen_rand_array(w128_t *array, int size) {
81 int i, j;
82 __m128i r, r1, r2, mask;
83 mask = _mm_set_epi32(MSK4, MSK3, MSK2, MSK1);
84
85 r1 = _mm_load_si128(&sfmt[N - 2].si);
86 r2 = _mm_load_si128(&sfmt[N - 1].si);
87 for (i = 0; i < N - POS1; i++) {
88 r = mm_recursion(&sfmt[i].si, &sfmt[i + POS1].si, r1, r2, mask);
89 _mm_store_si128(&array[i].si, r);
90 r1 = r2;
91 r2 = r;
92 }
93 for (; i < N; i++) {
94 r = mm_recursion(&sfmt[i].si, &array[i + POS1 - N].si, r1, r2, mask);
95 _mm_store_si128(&array[i].si, r);
96 r1 = r2;
97 r2 = r;
98 }
99 /* main loop */
100 for (; i < size - N; i++) {
101 r = mm_recursion(&array[i - N].si, &array[i + POS1 - N].si, r1, r2,
102 mask);
103 _mm_store_si128(&array[i].si, r);
104 r1 = r2;
105 r2 = r;
106 }
107 for (j = 0; j < 2 * N - size; j++) {
108 r = _mm_load_si128(&array[j + size - N].si);
109 _mm_store_si128(&sfmt[j].si, r);
110 }
111 for (; i < size; i++) {
112 r = mm_recursion(&array[i - N].si, &array[i + POS1 - N].si, r1, r2,
113 mask);
114 _mm_store_si128(&array[i].si, r);
115 _mm_store_si128(&sfmt[j++].si, r);
116 r1 = r2;
117 r2 = r;
118 }
119 }
120
35 #endif 121 #endif
36 #endif