
    J/Ph                         d dl Zd dlmZ d dlmZmZ d Zd Z G d de          Z	e
dk    r ej                     dS dS )	    N)cuda)unittestCUDATestCasec                 V    t          |j        d                   D ]}| |         ||<   d S Nr   )rangeshapeinpoutis      d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_slicing.pyfoor      s8    39Q<    QA     c                 v    t          j        d          }t          | |d d f         ||d d f                    d S )N   )r   gridcufoor
   s      r   copyr      s=    	!A	#ad)SAAAYr   c                       e Zd Zd Zd ZdS )TestCudaSlicingc                 F    t          j        dd          t                    a t          j        d          t                    }t          j        dt
          j                                      dd          }t          j	        |          } |d         ||           d S )	Nzvoid(int32[:], int32[:])T)devicezvoid(int32[:,:], int32[:,:])d   )dtype
   )r   r   )
r   jitr   r   r   nparangeint32reshape
zeros_like)selfcucopyr   r   s       r   test_slice_as_argz!TestCudaSlicing.test_slice_as_arg   s    A3DAAA#FF9899$??i28,,,44R<<mC  uc3r   c                     d}t          |          }t          j        t          |                    }t          j        |          |d d <   d S r   )r   r   device_arraylen	to_device)r#   Naarrs       r   test_assign_empty_slicez'TestCudaSlicing.test_assign_empty_slice   sE     !HHA''""AAAr   N)__name__
__module____qualname__r%   r-    r   r   r   r      s2             # # # # #r   r   __main__)numpyr   numbar   numba.cuda.testingr   r   r   r   r   r.   mainr1   r   r   <module>r7      s              5 5 5 5 5 5 5 5  
     
# # # # #l # # #( zHMOOOOO r   