
    sh                     n    S SK rS SKJr  S SKJrJr   " S S\5      r\S:X  a  \R                  " 5         gg)    N)cuda)unittestCUDATestCasec                       \ rS rSrS rSrg)TestFreeVar   c                    ^^ SSK Jn  SmUm[        R                  " S5      UU4S j5       n[        R
                  " SSS9nUS	   " US5        g
)zbMake sure we can compile the following kernel with freevar reference
in arguments to shared.array
r   )float32i   z(float32[::1], intp)c                 N   > [         R                  R                  TTS9nX!   X'   g)zDummy functiondtypeN)r   sharedarray)Aisdatanbtypesizes      x/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudapy/test_freevar.pyfoo%TestFreeVar.test_freevar.<locals>.foo   s,     KK%%d,2 & 4E8AD       r
   r   )   r   N)numbar
   r   jitnparange)selfr
   r   r   r   r   s       @@r   test_freevarTestFreeVar.test_freevar   sM     	"	(	)	 
*	 IIay)D	!Qr    N)__name__
__module____qualname____firstlineno__r    __static_attributes__r"   r   r   r   r      s    r   r   __main__)
numpyr   r   r   numba.cuda.testingr   r   r   r#   mainr"   r   r   <module>r,      s2      5, * zMMO r   