U
    9%e                     @   sH   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rDe	  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ZdtypeN)r   Zsharedarray)AiZsdataZnbtypesize c/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/numba/cuda/tests/cudapy/test_freevar.pyfoo   s    z%TestFreeVar.test_freevar.<locals>.foo   r   r   )   r   N)numbar   r   ZjitnpZ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   Znumba.cuda.testingr   r   r   r   mainr   r   r   r   <module>   s
   