a d'@s ddlZddlmZmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZeddZedd Zed d Zed d ZeddZeddZeddZeddZeddZeddZeddZeddZeddZed d!Zed"d#ZdS)$N)uint64uint32uint16uint8)register_jitable) UINT32_MAX UINT64_MAX UINT16_MAX UINT8_MAX) next_uint32 next_uint64cCsTt|}||d?O}||d?O}||d?O}||d?O}||d?O}||d?O}|S)N )r)maxmaskrg/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/numba/np/random/random_methods.pygen_masks      rcCsF|dkr|||fS|s$t|}d}n|dL}|d8}|d@dk||fS)Nrr )r )bitgenoffrngbcntbufrrrbuffered_bounded_bools rcCs0|st|}d}n|dL}|d8}t|||fS)Nrr )r rrrrrrrbuffered_uint8(s r!cCs0|st|}d}n|dL}|d8}t|||fS)Nr r)r rr rrrbuffered_uint164s r"c Cst|td}|dksJt|||\}}}t||}|d@}||krtt||}||krt|||\}}}t||}|d@}qZ|d?||fS)a? Generates a random unsigned 8 bit integer bounded within a given interval using Lemire's rejection. The buffer acts as storage for a 32 bit integer drawn from the associated BitGenerator so that multiple integers of smaller bitsize can be generated from a single draw of the BitGenerator. r r)rr!rr rrrrrng_exclnmleftover thresholdrrrbuffered_bounded_lemire_uint8Bs     r*c Cst|td}|dksJt|||\}}}t||}|d@}||krtt||}||krt|||\}}}t||}|d@}qZ|d?||fS)a@ Generates a random unsigned 16 bit integer bounded within a given interval using Lemire's rejection. The buffer acts as storage for a 32 bit integer drawn from the associated BitGenerator so that multiple integers of smaller bitsize can be generated from a single draw of the BitGenerator. r r)rr"rr r$rrrbuffered_bounded_lemire_uint16fs     r,cCszt|td}|dksJtt|t|}|d@}||krrt||}||krrtt|t|}|d@}qL|d?S)zr Generates a random unsigned 32 bit integer bounded within a given interval using Lemire's rejection. r r)rrr r)rrr%r'r(r)rrrbuffered_bounded_lemire_uint32s   r.cCst|td}|dksJt|}t|t|}||krjt||}||krjt|}t|t|}qH|td@}|d?}|td@}|d?} ||} ||| d?} | td@} | d?} | || 7} || | | d?}|S)zr Generates a random unsigned 64 bit integer bounded within a given interval using Lemire's rejection. r r-r)rr r)rrr%xr(r)Zx0x1Z rng_excl0Z rng_excl1Zw0tZw1Zw2m1rrrbounded_lemire_uint64s(      r4cCstj||d}|dkr0t|D] }|||<q n|dkr|dkrbt|D]}|t|||<qJqt|D]}|t||||<qlnL|dkrt|D]}|t|||<qn"t|D]}|t||||<q|S)z_ Returns a new array of given size with 64 bit integers bounded by given interval. dtyperr-r/)npemptyndindexr r.r r4rlowrsizer6outirrrrandom_bounded_uint64_fills  r?cCstj||d}|dkr0t|D] }|||<q nL|dkrZt|D]}|t|||<qBn"t|D]}|t||||<qd|S)z_ Returns a new array of given size with 32 bit integers bounded by given interval. r5rr-)r7r8r9r r.r:rrrrandom_bounded_uint32_fills r@c Csd}d}tj||d}|dkr8t|D] }|||<q(nh|dkrpt|D]"}t|||\} }}|| ||<qJn0t|D]$}t||||\} }}|| ||<qz|S)z_ Returns a new array of given size with 16 bit integers bounded by given interval. rr5r+)r7r8r9r"r, rr;rr<r6rrr=r>valrrrrandom_bounded_uint16_fills" rCc Csd}d}tj||d}|dkr8t|D] }|||<q(nh|dkrpt|D]"}t|||\} }}|| ||<qJn0t|D]$}t||||\} }}|| ||<qz|S)z^ Returns a new array of given size with 8 bit integers bounded by given interval. rr5r#)r7r8r9r!r*rArrrrandom_bounded_uint8_fills" rDc CsLd}d}tj||d}t|D]&}t|||||\} }}|| ||<q |S)z@ Returns a new array of given size with boolean values. rr5)r7r8r9rrArrrrandom_bounded_bool_fill/srEcCs||krtd|dkrjt|}|s0|td8}t|}|dkrHt|}||krXtd||krtdn ||krztd||krtddS)zS Check that low and high are within the bounds for the given datatype. zlow is out of boundsrr zhigh is out of boundsz*low is greater than high in given intervalN) ValueErrorr)r;highZendpointZ lower_boundZ upper_boundrrr_randint_arg_check=s"  rHcCs~|dkr dSt|}tt|}|dkrTtt||@}||krvtt||@}q8n"t||@}||krvt||@}q`t|S)Nrr-)rrr r )rZmax_valrvaluerrrrandom_interval\s  rJ) numpyr7ZnumbarrrrZnumba.core.extendingrZnumba.np.random._constantsrrr r Znumba.np.random.generator_corer r rrr!r"r*r,r.r4r?r@rCrDrErHrJrrrrsD      # #  "