o
    8ήc                     @   sX   d dl Zd dlmZmZmZ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float32int32void)unittestCUDATestCasec                   @   s   e Zd Zdd ZdS )TestCudaPy2Divc                 C   s   t ttd d  td d  td d  tdd }d}tj|tjd}tj|tjd}tj|tjd}|d |||| | t||k | t||k d S )Nc                 S   sF   t d}|| }tdt| }|| |< |d ||< |d ||< d S )N   gjt?g      ?)r   gridr   )yyAyBnumDataPointsikans r   Q/tmp/pip-target-vg8gfxp4/lib/python/numba/cuda/tests/cudapy/test_py2_div_issue.pypreCalc   s   
z2TestCudaPy2Div.test_py2_div_issue.<locals>.preCalc   )dtype)r	   r   )	r   jitr   r   r   npzeros
assertTrueall)selfr   r   r   r   r   r   r   r   test_py2_div_issue   s   *
z!TestCudaPy2Div.test_py2_div_issueN)__name__
__module____qualname__r   r   r   r   r   r      s    r   __main__)numpyr   numbar   r   r   r   numba.cuda.testingr   r   r   r   mainr   r   r   r   <module>   s    