U yʹj@s2UddlmZddlmZdZejed<dZejed<dZ ejed <d Z ejed <d Z ee fejd ddZ ee fejd ddZ ee fejd ddZee fejd ddZeddZee fejd ddZee fejd ddZeddZee fejd ddZee fejd d d!Zd"S)#)jit)corelyn< PHILOX_KEY_Al.v PHILOX_KEY_BlS$PHILOX_ROUND_AlW =PHILOX_ROUND_B )n_roundsc Csvt|D]^}t}t} ||} } t| | |A|A}t|| |A|A}| | }|| }|t}|t}q ||||fS)zV Run `n_rounds` rounds of Philox for state (c0, c1, c2, c3) and key (k0, k1). )tlZ static_rangerrZumulhirr) c0c1c2c3Zk0Zk1r _ABZ_c0Z_c2rU/var/www/html/bwave/python/venv/lib/python3.8/site-packages/triton/language/random.py philox_impls  rcCs|tj}|d?d@tj}|d@tj}|jtjdd}|jtjdd}|jtjdd}|jtjdd}t|||||||S)N lTZbitcast)tor Zuint64uint32r)seedr r rrr Zseed_hiZseed_lorrrphilox$s rcCst|||\}}}}|S)ar 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 offsets: The offsets to generate random numbers for. ) randint4x)roffsetr retrrrrrandint0s rcCs|d}t||||||S)aU 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)rrr Z_0rrrr@s rcCs4|jtjdd}d}t|dk| d|}||S)zq Numerically stable function to convert a random uint32 into a random float uniformly sampled in [0, 1). Trg=r r)rr Zint32where)xscalerrruint32_to_uniform_float]sr$cCs$|jtjdd}t|||}t|S)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. Tr)rr rrr$)rrr sourcerrrrandis  r&c CsP|jtjdd}t|||\}}}}t|}t|}t|} t|} ||| | fS)a Given a :code:`seed` scalar and an :code:`offsets` block, returns a 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. Tr)rr rrr$) roffsetsr i1i2Zi3i4u1u2u3u4rrrrand4xws r/cCsDtd|}d|}tdt|}|t||t|fS)zBox-Muller transformgHz>g-DT!@g)r maximumsqrtlogcossin)r+r,thrrrrpair_uniform_to_normals r7c Cs6t|||\}}}}t|}t|}t||\}}|S)a 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. )rr$r7) rrr r(r)rr+r,n1rrrrandns r9c Cs<t|||\}}}}t||\}}t||\} } ||| | fS)a  Given a :code:`seed` scalar and an :code:`offset` block, returns a 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/r7) rrr r+r,r-r.r8Zn2Zn3Zn4rrrrandn4xs r:N)Z runtime.jitrrr rZ constexpr__annotations__rrrZN_ROUNDS_DEFAULTrrrrr$r&r/r7r9r:rrrrs6