File: //lib/python3/dist-packages/numpy/random/_examples/numba/__pycache__/extending.cpython-310.pyc
o
6��a� � @ sD d dl Zd dlZd dlmZ d dlmZ e� Zejj Z
ejj Z dd� Z
eje
dd�ZdZd d
� Zej�e� �Zd d � Ze� Ze� ZejefksIJ �ejejksQJ �eed
d�Zeed�de� d�� eed
d�Zeed�de� d�� ej j Z ej j Z ejdd�dd� �Z!ee!dde j"�� ejdd�dd� �Z#e#ddde j"� dS )� N)�PCG64)�timeitc C s� t �| �}t| d d �D ]h}dt|� d }dt|� d }|| || }|dks/|dkrOdt|� d }dt|� d }|| || }|dks/|dks/t �dt �|� | �}|| |d| < d| d | k ru|| |d| d < q
|S )N� � g @g �?g g �)�np�empty�range�next_d�sqrt�log)�n�state�out�i�x1�x2�r2�f� r �H/usr/lib/python3/dist-packages/numpy/random/_examples/numba/extending.py�normals s
��r T)�nopythoni' c C s
t tt�S )N)�normalsjr �
state_addrr r r r � numbacall s
r c C s t jtd�S )N)�size)�rg�normalr r r r r � numpycall% s r i� )�numberz.2fz
secs for z% PCG64 (Numba/PCG64) gaussian randomsz% PCG64 (NumPy/PCG64) gaussian randomsc C sx || }}||d? O }||d? O }||d? O }||d? O }||d? O }t |�|@ }||kr8t |�|@ }||ks.| | S )Nr r � � � )�next_u32)�lb�ubr
�mask�delta�valr r r � bounded_uint8 s �r) iC iC�$ c C s2 t j|t jd�}t|�D ]
}t| ||�||< q d S )N)�dtype)r r �uint32r r) )r$ r% r r
r r r r r �
bounded_uintsK s �r, i��� )$�numpyr �numba�nb� numpy.randomr r �bit_gen�cffi� next_doubler �
state_addressr r �jitr r r �random� Generatorr r �r1r �shape�t1�print�t2�ctypes� next_uint32r# r
� ctypes_stater) �valuer, r r r r �<module> s: