o
    Y۷i=                     @   s^   d dl mZmZ d dlmZ d dlmZmZ edG dd dejZ	e
dkr-e  dS dS )	    )float64uint32)compile_ptx)skip_on_cudasimunittestz(Compilation unsupported in the simulatorc                   @   s   e Zd Zdd ZdS )TestCudaMandelc                 C   s.   dd }t ttttt t t f}t||dd dS )z+Just make sure we can compile this
        c                 S   s   || | }|| | }	| | }
| | }||
|  }|||	  }t ||}d}t|D ]}|| | }|j|j |j|j  dkrE|  S q+|S )Ny                   )complexrangerealimag)tidmin_xmax_xmin_ymax_ywidthheightiterspixel_size_xpixel_size_yxyr   r   czi r   Y/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/cuda/tests/cudapy/test_mandel.pymandel   s   
z*TestCudaMandel.test_mandel.<locals>.mandelT)deviceN)r   r   r   )selfr   argsr   r   r   test_mandel   s
   
zTestCudaMandel.test_mandelN)__name__
__module____qualname__r"   r   r   r   r   r      s    r   __main__N)numbar   r   numba.cuda.compilerr   numba.cuda.testingr   r   TestCaser   r#   mainr   r   r   r   <module>   s    