o
    i                     @   sL   d dl Zd dlmZ d dlmZmZ G dd deZedkr$e	  dS dS )    N)cuda)unittestCUDATestCasec                   @   s   e Zd Zdd ZdS )TestFreeVarc                    sL   ddl m} d| td fdd}tjd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                    s    t jj d}|| | |< dS )zDummy functiondtypeN)r   sharedarray)Aisdatanbtypesize b/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/cuda/tests/cudapy/test_freevar.pyfoo   s   z%TestFreeVar.test_freevar.<locals>.foo   r   r   )   r   N)numbar   r   jitnparange)selfr   r   r   r   r   r   test_freevar   s   zTestFreeVar.test_freevarN)__name__
__module____qualname__r   r   r   r   r   r      s    r   __main__)
numpyr   r   r   numba.cuda.testingr   r   r   r   mainr   r   r   r   <module>   s    