o
    i                      @   s\   d dl Zd dlmZ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float64int32void)unittestCUDATestCasec                   @   s   e Zd Zdd Zdd ZdS )TestCudaIDivc                 C   t   t ttd d d d f ttdd }tjdtjd}t |}|d |dd | }| 	t
|dk d S )Nc                 S   4   t |D ]}t |D ]}| ||f  d  < q
qd S Ng       @rangegridl_xl_yxy r   _/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/cuda/tests/cudapy/test_idiv.pydiv	   
   z*TestCudaIDiv.test_inplace_div.<locals>.div   r   dtype   r   r         ?)r   jitr   r   r   npones	to_devicecopy_to_host
assertTrueall)selfr   r   r   r   r   r   r   test_inplace_div       

zTestCudaIDiv.test_inplace_divc                 C   r
   )Nc                 S   r   r   r   r   r   r   r   
div_double   r   z8TestCudaIDiv.test_inplace_div_double.<locals>.div_doubler   r   r   r   r   )r   r    r   r   r   r!   r"   r#   r$   r%   r&   )r'   r*   r   r   r   r   r   r   test_inplace_div_double   r)   z$TestCudaIDiv.test_inplace_div_doubleN)__name__
__module____qualname__r(   r+   r   r   r   r   r	      s    r	   __main__)numpyr!   numbar   r   r   r   r   numba.cuda.testingr   r   r	   r,   mainr   r   r   r   <module>   s    