
    J/Phh                         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                     dS dS )    N)cudaint32float32)unittestCUDATestCased   c                    t           j                            t          t                    }t          j        d          }|dk    rt          t                    D ]}|||<   t          j                     ||         | |<   d S )N   r   )r   sharedarrayNr   gridrangesyncthreads)arysmijs       d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_globals.pysimple_smemr      so    			1e	$	$B	!AAvvq 	 	ABqEEUCFFF    
      c                     t          j        d          \  }}t           j                            t          t
          ft                    }|dz   |dz   z  |||f<   t          j                     |||f         | ||f<   d S )N   r
   )r   r   r   r   S0S1r   r   )r   r   r   r   s       r   coop_smem2dr      sp    9Q<<DAq			B8W	-	-BA!a% Bq!tH1a4C1IIIr   c                   4    e Zd Zd Zej        d             ZdS )TestCudaTestGlobalc           	      F    t          j        d          t                    }d}t          j        |t          j                  } |d|f         |           |                     t          j        |t          j        |t          j                  k                         dS )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   s     .48,--k::huBH---E3sbiRX&F&F&FFGGHHHHHr   c                     t          j        d          t                    }d}t          j        |t          j                  } |d|f         |           t          j        |          }t          |j        d                   D ]<}t          |j        d                   D ]}t          |dz             |dz   z  |||f<    =| 
                    t          j        ||                     dS )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     248011+>>huBJ///E3mC  sy|$$ 	3 	3A39Q<(( 3 3!!a%LLAE2AqD		3C--.....r   N)__name__
__module____qualname__r-   r   SkipTestr4    r   r   r    r       sE        	I 	I 	I / / / / /r   r    __main__)numpyr%   numbar   r   r   numba.cuda.testingr   r   r   r   r   r   r   r    r5   mainr9   r   r   <module>r?      s        & & & & & & & & & & 5 5 5 5 5 5 5 5     / / / / / / / /: zHMOOOOO r   