σ \ K]c@@s¬ddlmZmZddlZddlmZmZmZmZm Z m Z m Z m Z ddl Zejdej fdej fgdeƒZe eƒZe d„ƒZe d„ƒZe d „ƒZe d ƒe d ƒfZe d „ƒZe d „ƒZe d„ƒZe d„ƒZe d„ƒZejdejƒZejdejƒZe d„ƒZe d„ƒZ e d„ƒZ!ddd„Z"ddd„Z#dS(i(tprint_functiontabsolute_importN(tcudatfloat32tfloat64tuint32tint64tuint64t from_dtypetjitts0ts1taligncC@s˜t|ƒ}t|ƒ}|tdƒ}||tdƒ?Atdƒ}||tdƒ?Atdƒ}||tdƒ?A}|||d<|||d|td|ƒ?BS(sLeft rotate x by k bits.i@(RR(txtk((s0lib/python2.7/site-packages/numba/cuda/random.pytrotl?s  cC@s˜t|ƒ}||d}||d}||}||N}tt|tdƒƒƒ|A|tdƒ>A||d@r|||dN}|||dN}nt||ƒqDWq1W|||d<|||dS(s7Convert uint64 to float64 value in the range [0.0, 1.0)i ii5(RRR(R((s0lib/python2.7/site-packages/numba/cuda/random.pytuint64_to_unit_float64ys cC@st|ƒ}tt|ƒƒS(s7Convert uint64 to float32 value in the range [0.0, 1.0)(RRR(R((s0lib/python2.7/site-packages/numba/cuda/random.pytuint64_to_unit_float32€s cC@st|ƒ}tt||ƒƒS(sώReturn a float32 in range [0.0, 1.0) and advance ``states[index]``. :type states: 1D array, dtype=xoroshiro128p_dtype :param states: array of RNG states :type index: int64 :param index: offset in states to update :rtype: float32 (RRR(R R((s0lib/python2.7/site-packages/numba/cuda/random.pytxoroshiro128p_uniform_float32‡s cC@st|ƒ}tt||ƒƒS(sώReturn a float64 in range [0.0, 1.0) and advance ``states[index]``. :type states: 1D array, dtype=xoroshiro128p_dtype :param states: array of RNG states :type index: int64 :param index: offset in states to update :rtype: float64 (RRR(R R((s0lib/python2.7/site-packages/numba/cuda/random.pytxoroshiro128p_uniform_float64•s icC@sbt|ƒ}t||ƒ}t||ƒ}tjtdƒ tj|ƒƒtjt|ƒ}|S(s–Return a normally distributed float32 and advance ``states[index]``. The return value is drawn from a Gaussian of mean=0 and sigma=1 using the Box-Muller transform. This advances the RNG sequence by two steps. :type states: 1D array, dtype=xoroshiro128p_dtype :param states: array of RNG states :type index: int64 :param index: offset in states to update :rtype: float32 g@(RRtmathtsqrtRtlogtcostTWO_PI_FLOAT32(R Rtu1tu2tz0((s0lib/python2.7/site-packages/numba/cuda/random.pytxoroshiro128p_normal_float32§s 4cC@sbt|ƒ}t||ƒ}t||ƒ}tjtdƒ tj|ƒƒtjt|ƒ}|S(s–Return a normally distributed float32 and advance ``states[index]``. The return value is drawn from a Gaussian of mean=0 and sigma=1 using the Box-Muller transform. This advances the RNG sequence by two steps. :type states: 1D array, dtype=xoroshiro128p_dtype :param states: array of RNG states :type index: int64 :param index: offset in states to update :rtype: float64 g@(RRR R!RR"R#tTWO_PI_FLOAT64(R RR%R&R'((s0lib/python2.7/site-packages/numba/cuda/random.pytxoroshiro128p_normal_float64Ώs 4cC@s₯|jd}t|ƒ}t|ƒ}|dkr‘t|d|ƒx!t|ƒD]}t|dƒqNWx9td|ƒD]%}||d||s* : -