~iSSKJr SSKJr SSKJr \R "S5r\\4S\R 4Sjj5r\\4S\R 4Sjj5r \\4S\R 4S jj5r \\4S\R 4S jj5r \S 5r \\4S\R 4S jj5r \\4S\R 4S jj5r\S5r\\4S\R 4Sjj5r\\4S\R 4Sjj5rg))jit)core)math n_roundsc*UR[R:Xa SnSnSn Sn O:[R"UR[R:HS5 SnSnSn S n [R "U5Hn U n U n Xp[ R"X5U- U- n[ R"X5U- U- n[R"XS S 9n[R"XS S 9n[R"XGS S 9n[R"XXS S 9nM XX#4$) 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) dtypetluint32 static_assertuint64 static_rangerumulhimuladd)c0c1c2c3k0k1r PHILOX_KEY_A PHILOX_KEY_BPHILOX_ROUND_APHILOX_ROUND_B_AB_c0_c2s S/home/mande/repo/quber/.venv/lib/python3.13/site-packages/triton/language/random.py philox_implr$ s  xx299%/ %/ '1'1 RYY.0TU%7 %7 '9'9 __X &  S [[ 2 % * [[ 2 % * VVAe 4 VVAe 4 VVB > VVB >' 2>c [R"U5n[R"URR 55 UR [R 5n[R"U5n[R"U5n[R"U5n[R"U5n[R"URR5S:XaX[RnUS- S-R [R5nUS-R [R5nOj[R"[R"URR5S:HS5 [R n[R"SSUS9nUnUR USS 9nUR USS 9nUR USS 9nUR USS 9n[XX4XU5$) N l@z bitwidth not supported in philox)r)r Tbitcast) r to_tensorrr is_inttor constexprprimitive_bitwidthr fullr$) seedrrrrr int_dtypeseed_hiseed_los r#philoxr6-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 BBr%c([XU5un nU$)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)r2offsetrretrs r#randintr;HsT84LCAq Jr%c UR[R5nUS-n[R"URR 5S:a#US- R[R5nOUn[ XXTXB5$)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. r)r')r.r r r/r r0r6)r2r9r offset_lo_0 offset_his r#r8r8Xsf "))$I QB ||FLL334r9r\%%bii0  $9" ??r%c[R"UR[R:H5(d6[R"UR[R:H5(a!UR [RSS9nSnO[R "[R"UR[R:H5=(d1 [R"UR[R:H55 UR [RSS9nSn[R"US:U*S- U5nX-$)zg Numerically stable function to convert a random uint into a random float uniformly sampled in [0, 1). Tr*g=g;r)r) r r/r r int32r.rrint64where)xscales r#uint_to_uniform_floatrF~s ||AGGryy())R\\!''RXX:M-N-N DD4D (  agg&:;`r||AGGWYW_W_L_?`a DD4D (& QQ"A 9r%c0[XU5n[U5$)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;rF)r2r9rsources r#randrIsT8 ,F  ((r%c[XU5up4pV[U5n[U5n[U5n [U5n XxX4$)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. )r8rF) r2offsetsri1i2i3i4u1u2u3u4s r#rand4xrTsFth7NBB r "B r "B r "B r "B 2>r%c[R"SU5nSU-n[R"S[R"U5-5nU[R "U5-U[R "U5-4$)zBox-Muller transformgHz>g-DT!@g)r maximumrsqrtlogcossin)rPrQthrs r#pair_uniform_to_normalr]s^ FB B R B $"%&A txx| Q"- --r%cl[XU5up4 n[U5n[U5n[Xg5upU$)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. )r8rFr]) r2r9rrLrMrrPrQn1s r#randnr`s;T84LBAq r "B r "B "2 *EB Ir%c\[XU5up4pV[X45upx[XV5upXxX4$)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. )rTr]) r2r9rrPrQrRrSr_n2n3n4s r#randn4xres6D(3NBB #B +FB #B +FB 2>r%N) runtime.jitrrr rr/N_ROUNDS_DEFAULTr$r6r;r8rFrIrTr]r`rer%r#rjs]<<#AQ",,@:JC2<<CC43C BLL  5E@bll@@J&0@ ) ) )3C BLL  *..1A ",,  3C BLL  r%