
    xKg                     l    d dl Zd dlmZ d dlmZmZ  G d de      Zedk(  r ej                          yy)    N)cuda)unittestCUDATestCasec                       e Zd Zd Zy)TestFreeVarc                     ddl m} d|t        j                  d      fd       }t	        j
                  dd      } |d	   |d       y
)zrMake sure we can compile the following kernel with freevar reference
        in arguments to shared.array
        r   )float32i   z(float32[::1], intp)c                 X    t         j                  j                        }||   | |<   y)zDummy functiondtypeN)r   sharedarray)Aisdatanbtypesizes      h/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_freevar.pyfooz%TestFreeVar.test_freevar.<locals>.foo   s0     KK%%d,2 & 4E8AaD       r	   r   )   r   N)numbar	   r   jitnparange)selfr	   r   r   r   r   s       @@r   test_freevarzTestFreeVar.test_freevar   sO     	"	(	)	 
*	 IIay)D	!Qr   N)__name__
__module____qualname__r    r   r   r   r      s    r   r   __main__)
numpyr   r   r   numba.cuda.testingr   r   r   r   mainr"   r   r   <module>r'      s4      5, * zHMMO r   