libstdc++-v3/config/cpu/i486/opt/ext/opt_random.h

   1 // Optimizations for random number extensions, x86 version -*- C++ -*-
   2
   3 // Copyright (C) 2012-2013 Free Software Foundation, Inc.
   4 //
   5 // This file is part of the GNU ISO C++ Library.  This library is free
   6 // software; you can redistribute it and/or modify it under the
   7 // terms of the GNU General Public License as published by the
   8 // Free Software Foundation; either version 3, or (at your option)
   9 // any later version.
  10
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // Under Section 7 of GPL version 3, you are granted additional
  17 // permissions described in the GCC Runtime Library Exception, version
  18 // 3.1, as published by the Free Software Foundation.
  19
  20 // You should have received a copy of the GNU General Public License and
  21 // a copy of the GCC Runtime Library Exception along with this program;
  22 // see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
  23 // <http://www.gnu.org/licenses/>.
  24
  25 /** @file ext/random.tcc
  26  *  This is an internal header file, included by other library headers.
  27  *  Do not attempt to use it directly. @headername{ext/random}
  28  */
  29
  30 #ifndef _EXT_OPT_RANDOM_H
  31 #define _EXT_OPT_RANDOM_H 1
  32
  33 #pragma GCC system_header
  34
  35 #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
  36
  37 #ifdef __SSE2__
  38
  39 namespace __gnu_cxx _GLIBCXX_VISIBILITY(default)
  40 {
  41 _GLIBCXX_BEGIN_NAMESPACE_VERSION
  42
  43   namespace {
  44
  45     template<size_t __sl1, size_t __sl2, size_t __sr1, size_t __sr2,
  46              uint32_t __msk1, uint32_t __msk2, uint32_t __msk3, uint32_t __msk4>
  47       inline __m128i __sse2_recursion(__m128i __a, __m128i __b,
  48                                       __m128i __c, __m128i __d)
  49       {
  50         __m128i __y = _mm_srli_epi32(__b, __sr1);
  51         __m128i __z = _mm_srli_si128(__c, __sr2);
  52         __m128i __v = _mm_slli_epi32(__d, __sl1);
  53         __z = _mm_xor_si128(__z, __a);
  54         __z = _mm_xor_si128(__z, __v);
  55         __m128i __x = _mm_slli_si128(__a, __sl2);
  56         __y = _mm_and_si128(__y, _mm_set_epi32(__msk4, __msk3, __msk2, __msk1));
  57         __z = _mm_xor_si128(__z, __x);
  58         return _mm_xor_si128(__z, __y);
  59       }
  60
  61   }
  62
  63
  64 #define _GLIBCXX_OPT_HAVE_RANDOM_SFMT_GEN_READ  1
  65   template<typename _UIntType, size_t __m,
  66            size_t __pos1, size_t __sl1, size_t __sl2,
  67            size_t __sr1, size_t __sr2,
  68            uint32_t __msk1, uint32_t __msk2,
  69            uint32_t __msk3, uint32_t __msk4,
  70            uint32_t __parity1, uint32_t __parity2,
  71            uint32_t __parity3, uint32_t __parity4>
  72     void simd_fast_mersenne_twister_engine<_UIntType, __m,
  73                                            __pos1, __sl1, __sl2, __sr1, __sr2,
  74                                            __msk1, __msk2, __msk3, __msk4,
  75                                            __parity1, __parity2, __parity3,
  76                                            __parity4>::
  77     _M_gen_rand(void)
  78     {
  79       __m128i __r1 = _mm_load_si128(&_M_state[_M_nstate - 2]);
  80       __m128i __r2 = _mm_load_si128(&_M_state[_M_nstate - 1]);
  81
  82       size_t __i;
  83       for (__i = 0; __i < _M_nstate - __pos1; ++__i)
  84         {
  85           __m128i __r = __sse2_recursion<__sl1, __sl2, __sr1, __sr2,
  86                                          __msk1, __msk2, __msk3, __msk4>
  87             (_M_state[__i], _M_state[__i + __pos1], __r1, __r2);
  88           _mm_store_si128(&_M_state[__i], __r);
  89           __r1 = __r2;
  90           __r2 = __r;
  91         }
  92       for (; __i < _M_nstate; ++__i)
  93         {
  94           __m128i __r = __sse2_recursion<__sl1, __sl2, __sr1, __sr2,
  95                                          __msk1, __msk2, __msk3, __msk4>
  96             (_M_state[__i], _M_state[__i + __pos1 - _M_nstate], __r1, __r2);
  97           _mm_store_si128(&_M_state[__i], __r);
  98           __r1 = __r2;
  99           __r2 = __r;
 100         }
 101
 102       _M_pos = 0;
 103     }
 104
 105
 106 #define _GLIBCXX_OPT_HAVE_RANDOM_SFMT_OPERATOREQUAL     1
 107   template<typename _UIntType, size_t __m,
 108            size_t __pos1, size_t __sl1, size_t __sl2,
 109            size_t __sr1, size_t __sr2,
 110            uint32_t __msk1, uint32_t __msk2,
 111            uint32_t __msk3, uint32_t __msk4,
 112            uint32_t __parity1, uint32_t __parity2,
 113            uint32_t __parity3, uint32_t __parity4>
 114     bool
 115     operator==(const __gnu_cxx::simd_fast_mersenne_twister_engine<_UIntType,
 116                __m, __pos1, __sl1, __sl2, __sr1, __sr2,
 117                __msk1, __msk2, __msk3, __msk4,
 118                __parity1, __parity2, __parity3, __parity4>& __lhs,
 119                const __gnu_cxx::simd_fast_mersenne_twister_engine<_UIntType,
 120                __m, __pos1, __sl1, __sl2, __sr1, __sr2,
 121                __msk1, __msk2, __msk3, __msk4,
 122                __parity1, __parity2, __parity3, __parity4>& __rhs)
 123     {
 124       __m128i __res = _mm_cmpeq_epi8(__lhs._M_state[0], __rhs._M_state[0]);
 125       for (size_t __i = 1; __i < __lhs._M_nstate; ++__i)
 126         __res = _mm_and_si128(__res, _mm_cmpeq_epi8(__lhs._M_state[__i],
 127                                                     __rhs._M_state[__i]));
 128       return (_mm_movemask_epi8(__res) == 0xffff
 129               && __lhs._M_pos == __rhs._M_pos);
 130     }
 131
 132
 133 _GLIBCXX_END_NAMESPACE_VERSION
 134 } // namespace
 135
 136 #endif // __SSE2__
 137
 138 #endif // __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 139
 140 #endif // _EXT_OPT_RANDOM_H