
    shY              	           S SK r S SKJrJrJrJrJr  \\\" S5      \" S5       " S S\5      5       5       5       5       r\S:X  a  \ R                  " 5         gg)    N)CUDATestCaseskip_on_cudasimskip_if_cudadevrt_missingskip_unless_cc_60skip_if_mvc_enabledzCG not supported with MVCz4cudasim doesn't support cuda import at non-top-levelc                       \ rS rSrS rSrg)TestCooperativeGroups
   c                   ^
 SSK Jm
Jn  SS KnUS S 2S S S24   4nT
R	                  U5      U
4S j5       nUR                  SUR                  S9nSnUR                  S   U-  nUR                  U   R                  U5      nX:  a  U R                  S5        XGU4   " U5        UR                  UR                  S	5      S
5      R                  n	UR                  R                  XY5        g )Nr   )cudaint32   c                   > TR                  S5      nTR                  R                  5       nU R                  S   nU R                  S   n[	        SU5       H*  nXA-
  S-
  nXS-
  U4   S-   XU4'   UR                  5         M,     g )Nr   r   )gridcg	this_gridshaperangesync)Mcolgrowscolsrowoppositer   s          y/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/doc_examples/test_cg.pysequential_rows@TestCooperativeGroups.test_ex_grid_sync.<locals>.sequential_rows   s~    ))A,C!!#A771:D771:DQ~:>a 12Q6s(  &    )   r!   )dtype    z0Device does not support a large enough coop gridr!   )r!   r   )numbar   r   numpyjitzerosr   	overloadsmax_cooperative_grid_blocksskipTesttilearangeTtestingassert_equal)selfr   npsigr   Ablockdimgriddimmb	referencer   s             @r   test_ex_grid_sync'TestCooperativeGroups.test_ex_grid_sync   s    %QssU|o	#	 
	$ HH\H2 ''!*(
 &&s+GGQ<MMLM 	)*1- GGBIIdOY799	


-r     N)__name__
__module____qualname____firstlineno__r8   __static_attributes__r:   r    r   r	   r	   
   s    
:.r    r	   __main__)
unittestnumba.cuda.testingr   r   r   r   r   r	   r;   mainr:   r    r   <module>rD      sm    5 5
 01GH;.L ;. I 2  ;.| zMMO r    