
    xKgh                         d dl Zd dlmZmZmZ d dlmZmZ dZ	d Z
dZdZd Z G d	 d
e      Zedk(  r ej                           yy)    N)cudaint32float32)unittestCUDATestCased   c                     t         j                  j                  t        t              }t        j
                  d      }|dk(  rt        t              D ]  }|||<   	 t        j                          ||   | |<   y )N   r   )r   sharedarrayNr   gridrangesyncthreads)arysmijs       h/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_globals.pysimple_smemr      s\    			1e	$B		!AAvqABqE UCF    
      c                     t        j                  d      \  }}t         j                  j                  t        t
        ft              }|dz   |dz   z  |||f<   t        j                          |||f   | ||f<   y )N   r
   )r   r   r   r   S0S1r   r   )r   r   r   r   s       r   coop_smem2dr      sh    99Q<DAq			B8W	-BA!a% Bq!tH1a4C1Ir   c                   6    e Zd Zd Zej
                  d        Zy)TestCudaTestGlobalc           	      >    t        j                  d      t              }d}t        j                  |t        j
                        } |d|f   |       | j                  t        j                  |t        j                  |t        j
                        k(               y)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_constz(TestCudaTestGlobal.test_global_int_const   so     .488,-k:hhuBHH-E3sbiiRXX&FFGHr   c                     t        j                  d      t              }d}t        j                  |t        j
                        } |d|f   |       t        j                  |      }t        |j                  d         D ]8  }t        |j                  d         D ]  }t        |dz         |dz   z  |||f<    : | j                  t        j                  ||             y)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+   r0   r   expr   r   s          r   test_global_tuple_constz*TestCudaTestGlobal.test_global_tuple_const*   s     248801+>hhuBJJ/E3mmC syy|$A399Q<(!!a%LAE2AqD	 ) % 	C-.r   N)__name__
__module____qualname__r-   r   SkipTestr4    r   r   r    r       s"    	I / /r   r    __main__)numpyr%   numbar   r   r   numba.cuda.testingr   r   r   r   r   r   r   r    r5   mainr9   r   r   <module>r?      sR     & & 5 / /: zHMMO r   