
    shh                         S SK rS SKJrJrJr  S SKJrJr  Sr	S r
SrSrS r " S	 S
\5      r\S:X  a  \R                   " 5         gg)    N)cudaint32float32)unittestCUDATestCased   c                     [         R                  R                  [        [        5      n[         R
                  " S5      nUS:X  a  [        [        5       H  nX1U'   M	     [         R                  " 5         X   X'   g )N   r   )r   sharedarrayNr   gridrangesyncthreads)arysmijs       x/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudapy/test_globals.pysimple_smemr      sV    			1e	$B		!AAvqAqE UCF    
      c                     [         R                  " S5      u  p[         R                  R                  [        [
        4[        5      nUS-   US-   -  X1U4'   [         R                  " 5         X1U4   XU4'   g )N   r
   )r   r   r   r   S0S1r   r   )r   r   r   r   s       r   coop_smem2dr      s`    99Q<DA			B8W	-BA!a% B!tHa4C1Ir   c                   >    \ rS rSrS r\R                  S 5       rSrg)TestCudaTestGlobal   c           
      :   [         R                  " S5      " [        5      nSn[        R                  " U[        R
                  S9nUSU4   " U5        U R                  [        R                  " U[        R                  " U[        R
                  S9:H  5      5        g)zTest simple_smem
        zvoid(int32[:])r   dtyper
   N)	r   jitr   npemptyr   
assertTrueallarange)selfcompilednelemr   s       r   test_global_int_const(TestCudaTestGlobal.test_global_int_const   sk     88,-k:hhuBHH-E3sbiiRXX&FFGHr   c                    [         R                  " S5      " [        5      nSn[        R                  " U[        R
                  S9nUSU4   " U5        [        R                  " U5      n[        UR                  S   5       H:  n[        UR                  S   5       H  n[        US-   5      US-   -  XEU4'   M     M<     U R                  [        R                  " X45      5        g)zTest coop_smem2d
        zvoid(float32[:,:]))r   r   r#   r
   r   N)r   r%   r   r&   r'   r   
empty_liker   shapefloatr(   allclose)r+   r,   r2   r   expr   r   s          r   test_global_tuple_const*TestCudaTestGlobal.test_global_tuple_const*   s     8801+>hhuBJJ/E3mmC syy|$A399Q<(!!a%LAE2qD	 ) % 	C-.r    N)	__name__
__module____qualname____firstlineno__r.   r   SkipTestr6   __static_attributes__r8   r   r   r    r       s"    	I / /r   r    __main__)numpyr&   numbar   r   r   numba.cuda.testingr   r   r   r   r   r   r   r    r9   mainr8   r   r   <module>rD      sP     & & 5 / /: zMMO r   