o i(@sddlZddlmZmZmZmZmZmZmZm Z m Z ddl Z e j de jfde jfgddZe eZejZZej Ze eeeddd Ze eeedd d Ze eeedd d Ze eeedddZe eeedddZe eeedddZe eeedddZe eeedddZe dejZe dejZe eeedddZe eeedddZ e eeedddZ!d#dd Z"d#d!d"Z#dS)$N) configcudafloat32float64uint32int64uint64 from_dtypejits0s1T)align)ZforceobjZloopliftZnopythoncCsxt|}t|}|td}||td?Atd}||td?Atd}||td?A}|||d<|||d<d S) aUse SplitMix64 to generate an xoroshiro128p state from 64-bit seed. This ensures that manually set small seeds don't result in a predictable initial sequence from the random number generator. :type states: 1D array, dtype=xoroshiro128p_dtype :param states: array of RNG states :type index: uint64 :param index: offset in states to update :type seed: int64 :param seed: seed value to use when initializing state l|~fq le9z lb&&& r r N)rrr)statesindexseedzr`/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/cuda/random.pyinit_xoroshiro128p_state.s  rcCs(t|}t|}||>|td|?BS)zLeft rotate x by k bits.@)rr)xkrrrrotlHsrcCsxt|}||d}||d}||}||N}tt|td|A|td>A||d<tt|td||d<|S)zReturn the next random uint64 and advance the RNG in 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: uint64 r r 7$)rrrr)rrr r resultrrrxoroshiro128p_nextPs  *r cCst|}tdtdf}td}td}tdD]*}tdD]#}||tdt|>@r<|||dN}|||dN}t||qq|||d<|||d<d S) zAdvance the RNG in ``states[index]`` by 2**64 steps. :type states: 1D array, dtype=xoroshiro128p_dtype :param states: array of RNG states :type index: int64 :param index: offset in states to update lzKW`u l"U .XC rrr r N)rrrangerr )rrZjumpr r ibrrrxoroshiro128p_jumpfs     r&cCs,t|}|td?tdtdtd>S)z7Convert uint64 to float64 value in the range [0.0, 1.0) r"5)rrrrrrruint64_to_unit_float64s$r*cCst|}tt|S)z7Convert uint64 to float32 value in the range [0.0, 1.0))rrr*r)rrruint64_to_unit_float32s r+cCt|}tt||S)zReturn 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 )rr+r rrrrrxoroshiro128p_uniform_float32 r.cCr,)zReturn 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 )rr*r r-rrrxoroshiro128p_uniform_float64r/r0r!cCHt|}t||}t||}ttd t|tt|}|S)aReturn 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 @)rr.mathsqrtrlogcosTWO_PI_FLOAT32rru1u2Zz0rrrxoroshiro128p_normal_float32   (r;cCr1)aReturn 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 r2)rr.r3r4rr5r6TWO_PI_FLOAT64r8rrrxoroshiro128p_normal_float64r<r>cCsx|jd}t|}t|}|dkr8t|d|t|D]}t|dqtd|D]}||d||<t||q(dSdS)Nrr")shaperrr#r&)rrsubsequence_startn_r$rrrinit_xoroshiro128p_states_cpus     rCcCs.tj|jtd}t||||j||ddS)aInitialize RNG states on the GPU for parallel generators. This initializes the RNG states so that each state in the array corresponds subsequences in the separated by 2**64 steps from each other in the main sequence. Therefore, as long no CUDA thread requests more than 2**64 random numbers, all of the RNG states produced by this function are guaranteed to be independent. The subsequence_start parameter can be used to advance the first RNG state by a multiple of 2**64 steps. :type states: 1D DeviceNDArray, dtype=xoroshiro128p_dtype :param states: array of RNG states :type seed: uint64 :param seed: starting seed for list of generators )r?dtype)streamN)npemptyr?xoroshiro128p_dtyperCZcopy_to_device)rrr@rEZ states_cpurrrinit_xoroshiro128p_statess rIcCs"tj|t|d}t|||||S)a5Returns a new device array initialized for n random number generators. This initializes the RNG states so that each state in the array corresponds subsequences in the separated by 2**64 steps from each other in the main sequence. Therefore, as long no CUDA thread requests more than 2**64 random numbers, all of the RNG states produced by this function are guaranteed to be independent. The subsequence_start parameter can be used to advance the first RNG state by a multiple of 2**64 steps. :type n: int :param n: number of RNG states to create :type seed: uint64 :param seed: starting seed for list of generators :type subsequence_start: uint64 :param subsequence_start: :type stream: CUDA stream :param stream: stream to run initialization kernel on )rDrE)rZ device_arrayrHrI)rArr@rErrrrcreate_xoroshiro128p_states srJ)rr)$r3Znumbarrrrrrrr r numpyrFrDrHZxoroshiro128p_typeZENABLE_CUDASIMZ _forceobjZ _loopliftZ _nopythonrrr r&r*r+r.r0pir7r=r;r>rCrIrJrrrrsF,