+ jN^RIHt^RIHt^RIHt]P !^ 4t]]3RRll4t]]3RRll4t ]]3RR ll4t ]]3R R ll4t ]R 4t ]]3R Rll4t ]]3RRll4t]R4t]]3RRll4t]]3RRll4tR#))jit)core)mathc8V^8dQhR\P/#rn_roundstl constexpr)formats"W/home/luca/screen-checking/.venv/lib/python3.14/site-packages/triton/language/random.py __annotate__r s",,cdVP\P8Xd RpRpRp Rp M:\P!VP\P8HR4RpRpRp R p \P !V4Fp T p T p Yr\ P!W4V, V, p\ P!W4V, V, p\P!WR R 7p\P!WR R 7p\P!WGR R 7p\P!WXR R 7pK WW#3#) zN Run `n_rounds` rounds of Philox for state (c0, c1, c2, c3) and key (k0, k1). lyn<l.vlS$lW =z"dtype not supported in philox_impll|~fq l;' :=[ llB;: lW$* R F)sanitize_overflow) dtyper uint32 static_assertuint64 static_rangerumulhimuladd)c0c1c2c3k0k1r PHILOX_KEY_A PHILOX_KEY_BPHILOX_ROUND_APHILOX_ROUND_B_AB_c0_c2s&&&&&&& r philox_implr) s  xx299%/ %/ '1'1 RYY.0TU%7 %7 '9'9 __X &  S [[ 2 % * [[ 2 % * VVAe 4 VVAe 4 VVB > VVB >' 2>rc8V^8dQhR\P/#rr )r s"r rr.sCC2<<Crc ,\P!V4p\P!VPP 44VP \P 4p\P!V4p\P!V4p\P!V4p\P!V4p\P!VPP4^ 8Xde\PpV^ , R,P \P4pVR,P \P4pMm\P!\P!VPP4^@8HR4\P p\P!R^VR7pTpVP VRR7pVP VRR7pVP VRR7pVP VRR7p\WW4WV4#) lz bitwidth not supported in philox)rTbitcast)) r to_tensorrris_inttorr primitive_bitwidthrfullr)) seedrrrrr int_dtypeseed_hiseed_los &&&&&& r philoxr9-sq << DTZZ&&() 77299 D b B b B b B b B ||BHH//0B6II BJ*,00;*$((3 bhh&A&ABbHJlmII ''%)4 y$ 'B y$ 'B y$ 'B y$ 'B rrw BBrc8V^8dQhR\P/#rr )r s"r rrIs  BLL rc(\WV4wppV#)aU Given a :code:`seed` scalar and an :code:`offset` block, returns a single block of random :code:`int32`. If you need multiple streams of random numbers, using `randint4x` is likely to be faster than calling `randint` 4 times. :param seed: The seed for generating random numbers. :param offset: The offsets to generate random numbers for. ) randint4x)r5offsetrretr$s&&& r randintr?HsT84LCAq Jrc8V^8dQhR\P/#rr )r s"r rrYs@@bll@rcVP\P4pV^,p\P!VPP 4^ 8d(V^ , P\P4pMTp\ WWTWB4#)a9 Given a :code:`seed` scalar and an :code:`offset` block, returns four blocks of random :code:`int32`. This is the maximally efficient entry point to Triton's Philox pseudo-random number generator. :param seed: The seed for generating random numbers. :param offsets: The offsets to generate random numbers for. )r2r rr rr3r9)r5r=r offset_lo_0 offset_his&&& r r<r<Xsf "))$I QB ||FLL334r9r\%%bii0  $9" ??rc\P!VP\P8H4'g8\P!VP\P8H4'd%VP \PRR7pRpM\P !\P!VP\P8H4;'g2\P!VP\P8H44VP \PRR7pRp\P!V^8V(V4pW,#)zg Numerically stable function to convert a random uint into a random float uniformly sampled in [0, 1). Tr-g=g;) r r rrint32r2rrint64where)xscales& r uint_to_uniform_floatrK~s ||AGGryy())R\\!''RXX:M-N-N DD4D (  agg&:;``r||AGGWYW_W_L_?`a DD4D (& QAA 9rc8V^8dQhR\P/#rr )r s"r rrs ) ) )rc0\WV4p\V4#)z Given a :code:`seed` scalar and an :code:`offset` block, returns a block of random :code:`float32` in :math:`U(0, 1)`. :param seed: The seed for generating random numbers. :param offsets: The offsets to generate random numbers for. )r?rK)r5r=rsources&&& r randrOsT8 ,F  ((rc8V^8dQhR\P/#rr )r s"r rrs  BLL rc\WV4wr4rV\V4p\V4p\V4p \V4p WxW3#)z Given a :code:`seed` scalar and an :code:`offsets` block, returns 4 blocks of random :code:`float32` in :math:`U(0, 1)`. :param seed: The seed for generating random numbers. :param offsets: The offsets to generate random numbers for. )r<rK) r5offsetsri1i2i3i4u1u2u3u4s &&& r rand4xr[sFth7NBB r "B r "B r "B r "B 2>rc\P!RV4pRV,p\P!R\P!V4,4pV\P !V4,V\P !V4,3#)zBox-Muller transformgHz>g-DT!@g)r maximumrsqrtlogcossin)rWrXthrs&& r pair_uniform_to_normalrds^ FB B R B $"%&A txx| Q"- --rc8V^8dQhR\P/#rr )r s"r rrs  ",, rcl\WV4wr4p\V4p\V4p\Wg4wrV#)z Given a :code:`seed` scalar and an :code:`offset` block, returns a block of random :code:`float32` in :math:`\mathcal{N}(0, 1)`. :param seed: The seed for generating random numbers. :param offsets: The offsets to generate random numbers for. )r<rKrd) r5r=rrSrTr$rWrXn1s &&& r randnrhs;T84LBAq r "B r "B "2 *EB Irc8V^8dQhR\P/#rr )r s"r rrs  BLL rc\\WV4wr4rV\W44wrx\WV4wrWxW3#)z Given a :code:`seed` scalar and an :code:`offset` block, returns 4 blocks of random :code:`float32` in :math:`\mathcal{N}(0, 1)`. :param seed: The seed for generating random numbers. :param offsets: The offsets to generate random numbers for. )r[rd) r5r=rrWrXrYrZrgn2n3n4s &&& r randn4xrns6D(3NBB #B +FB #B +FB 2>rN) runtime.jitrrr rr N_ROUNDS_DEFAULTr)r9r?r<rKrOr[rdrhrnrr rss<<#AQ@:JCC43C  5E@@J&0@ ) )3C  *..1A  3C  r