
    J/Ph                     t    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	                     dS dS )    N)cuda)unittestCUDATestCasec                       e Zd Zd ZdS )TestFreeVarc                     ddl m} d|t          j        d          fd            }t	          j        dd          } |d	         |d           d
S )zrMake sure we can compile the following kernel with freevar reference
        in arguments to shared.array
        r   )float32i   z(float32[::1], intp)c                 `    t           j                                      }||         | |<   dS )zDummy functiondtypeN)r   sharedarray)Aisdatanbtypesizes      d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_freevar.pyfooz%TestFreeVar.test_freevar.<locals>.foo   s7     K%%d,2 & 4 4E8AaDDD       r	   r   )   r   N)numbar	   r   jitnparange)selfr	   r   r   r   r   s       @@r   test_freevarzTestFreeVar.test_freevar   s     	"!!!!!	(	)	)	 	 	 	 	 
*	)	 Iay)))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'      s              5 5 5 5 5 5 5 5    ,   * zHMOOOOO r   