
    xKg                     x    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                          yy)	    N)cuda)unittestCUDATestCasec                 N    t        |j                  d         D ]
  }| |   ||<    y Nr   )rangeshapeinpoutis      h/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_slicing.pyfoor      s&    399Q< QA !    c                 b    t        j                  d      }t        | |d d f   ||d d f          y )N   )r   gridcufoor
   s      r   copyr      s)    		!A	#ad)SAYr   c                       e Zd Zd Zd Zy)TestCudaSlicingc                 0    t        j                  dd      t              a t        j                  d      t              }t        j                  dt
        j                        j                  dd      }t        j                  |      } |d   ||       y )	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   so    A3DA#F989$?ii288,44R<mmC uc3r   c                     d}t        |      }t        j                  t        |            }t        j                  |      |d d  y r   )r   r   device_arraylen	to_device)r#   Naarrs       r   test_assign_empty_slicez'TestCudaSlicing.test_assign_empty_slice   s8     !HA'"Ar   N)__name__
__module____qualname__r%   r-    r   r   r   r      s     #r   r   __main__)numpyr   numbar   numba.cuda.testingr   r   r   r   r   r.   mainr1   r   r   <module>r7      s>      5
 
#l #( zHMMO r   