i(ddlZddlmZmZmZmZmZmZmZm Z m Z ddl Z e jde jfde jfgdZe eZej xZZej Ze eeedZe eeed Ze eeed Ze eeed Ze eeed Ze eeed Ze eeedZe eeedZe jdej8zZe j dej8zZe eeedZe eeedZ e eeedZ!ddZ"ddZ#y)N) configcudafloat32float64uint32int64uint64 from_dtypejits0s1T)align)forceobjloopliftnopythonct|}t|}|tdz}||tdz z tdz}||tdz z tdz}||tdz z }|||d<|||d<y ) 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)rr r)statesindexseedzs f/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/cuda/random.pyinit_xoroshiro128p_stater.s %LE $)rr9r?r@rrArBTWO_PI_FLOAT64rDs rxoroshiro128p_normal_float64rLrIrc|jd}t|}t|}|dk\rRt|d|t|D]}t |dtd|D]}||dz ||<t ||yy)Nrr*)shaper rr+r/)rrsubsequence_startn_r-s rinit_xoroshiro128p_states_cpurRs QA $rhs $$$4bhhryy 1D"))3DE%)+ 34--- I%% % i)i@A2i)i@,A,i)i@A*i)i@A4i)i@HAH i)i@.A. i)i@ EA Ei)i@ EA EAK(AK(i)i@A0i)i@A0i)i@*A*$52r