
    xKg                      |    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                          yy)    N)cudafloat32float64int32void)unittestCUDATestCasec                       e Zd Zd Zd Zy)TestCudaIDivc                 v   t        j                  t        t        d d d d f   t        t                    d        }t        j                  dt
        j                        }t        j                  |      } |d   |dd       |j                         }| j                  t        j                  |dk(               y )Nc                 b    t        |      D ]!  }t        |      D ]  }| ||fxx   dz  cc<    # y Ng       @rangegridl_xl_yxys        e/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_idiv.pydivz*TestCudaIDiv.test_inplace_div.<locals>.div	   0    3ZsAAJ#%J $         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    	$wq!t}eU3	4	& 
5	&
 GGF"**-~~a D	$1qCx()r   c                 v   t        j                  t        t        d d d d f   t        t                    d        }t        j                  dt
        j                        }t        j                  |      } |d   |dd       |j                         }| j                  t        j                  |dk(               y )Nc                 b    t        |      D ]!  }t        |      D ]  }| ||fxx   dz  cc<    # y 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    	$wq!t}eU3	4	& 
5	&
 GGF"**-~~a 
4q!$qCx()r   N)__name__
__module____qualname__r*   r.    r   r   r   r      s    **r   r   __main__)numpyr#   numbar   r   r   r   r   numba.cuda.testingr   r	   r   r/   mainr2   r   r   <module>r8      s7     5 5 5*< *< zHMMO r   