o
    Û¾i[  ã                   @   sL   d dl Z d dlmZ d dlmZmZ G dd„ deƒZedkr$e ¡  dS dS )é    N)Úcuda)ÚunittestÚCUDATestCasec                   @   s   e Zd Zdd„ ZdS )ÚTestCudaMonteCarloc                 C   s   t  d¡dd„ ƒ}dS )z+Just make sure we can compile this
        z=void(double[:], double[:], double, double, double, double[:])c                 S   sJ   t  d¡}||jd krd S || }| | t || ||  ¡ ||< d S )Né   r   )r   ÚgridÚshapeÚmathÚexp)ÚlastÚpathsÚdtÚc0Úc1ÚnormdistÚiÚnoise© r   ú[/home/ubuntu/.local/lib/python3.10/site-packages/numba/cuda/tests/cudapy/test_montecarlo.pyÚstep   s
   
&z0TestCudaMonteCarlo.test_montecarlo.<locals>.stepN)r   Újit)Úselfr   r   r   r   Útest_montecarlo   s   ÿz"TestCudaMonteCarlo.test_montecarloN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s    r   Ú__main__)	r	   Únumbar   Únumba.cuda.testingr   r   r   r   Úmainr   r   r   r   Ú<module>   s    ÿ