
    J/Ph                          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e
          Zedk    r e	j                     dS dS )    N)cudafloat32float64int32void)unittestCUDATestCasec                       e Zd Zd Zd ZdS )TestCudaIDivc                    t          j        t          t          d d d d f         t          t                              d             }t          j        dt
          j                  }t          j        |          } |d         |dd           |                                }| 	                    t          j
        |dk                         d S )Nc                 r    t          |          D ]&}t          |          D ]}| ||fxx         dz  cc<   'd S Ng       @rangegridl_xl_yxys        a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_idiv.pydivz*TestCudaIDiv.test_inplace_div.<locals>.div	   W    3ZZ & &s & &AAJJJ#%JJJJ&& &       r   dtype   r    r         ?)r   jitr   r   r   npones	to_devicecopy_to_host
assertTrueall)selfr   r   r   r   s        r   test_inplace_divzTestCudaIDiv.test_inplace_div   s    	$wqqq!!!t}eU33	4	4	& 	& 
5	4	&
 GF"*---~a  D	$1qCx(()))))r   c                    t          j        t          t          d d d d f         t          t                              d             }t          j        dt
          j                  }t          j        |          } |d         |dd           |                                }| 	                    t          j
        |dk                         d S )Nc                 r    t          |          D ]&}t          |          D ]}| ||fxx         dz  cc<   'd S r   r   r   s        r   
div_doublez8TestCudaIDiv.test_inplace_div_double.<locals>.div_double   r   r   r   r   r   r   r!   )r   r"   r   r   r   r#   r$   r%   r&   r'   r(   )r)   r-   r   r   r   s        r   test_inplace_div_doublez$TestCudaIDiv.test_inplace_div_double   s    	$wqqq!!!t}eU33	4	4	& 	& 
5	4	&
 GF"*---~a  
4q!$$$qCx(()))))r   N)__name__
__module____qualname__r*   r.    r   r   r   r      s2        * * ** * * * *r   r   __main__)numpyr#   numbar   r   r   r   r   numba.cuda.testingr   r	   r   r/   mainr2   r   r   <module>r8      s        5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5* * * * *< * * *< zHMOOOOO r   