
    sh                     x    S SK r S SKJrJr  \" S5       " S S\5      5       r\S:X  a  \ R                  " 5         gg)    N)CUDATestCaseskip_on_cudasimz4cudasim doesn't support cuda import at non-top-levelc                       \ rS rSrS rSrg)
TestRandom   c                 ,  ^^ SSK Jm  SSKJnJm  SS KnTR                  UU4S j5       nSu  pEnSu  pxn	Su  pnXx-  U	-  U
-  U-  U-  nU" USS	9nTR                  XEU4UR                  S
9nX:X4XxU	44   " X5        UR                  5       nU R                  UR                  U5      S5        U R                  UR                  U5      S5        U R                  UR                  US:*  5      5        U R                  UR                  US:  5      5        g )Nr   )cuda)create_xoroshiro128p_statesxoroshiro128p_uniform_float32c                 L  > TR                  S5      u  p#nTR                  S5      u  pVnXF-  U-  X5-  -   U-   n[        X@R                  S   U5       HO  n	[        X0R                  S   U5       H/  n
[        X R                  S   U5       H  nT" X5      X	X4'   M     M1     MQ     g )N   r         )gridgridsizerangeshape)arr
rng_statesstartxstartystartzstridexstrideystrideztidijkr	   r   s               }/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/doc_examples/test_random.py	random_3d-TestRandom.test_ex_3d_grid.<locals>.random_3d   s     &*YYq\"FF(,a(8%Gg #g-&2BCfLC 699Q<9vyy|W=A"699Q<A'DZ'UqG B > :    )i  i  i  )r   r   r   )   r$   r$   r   )seed)dtypeg\(\?gRQ?g      ?g        )numbar	   numba.cuda.randomr
   r   numpyjitdevice_arrayfloat32copy_to_hostassertGreatermean
assertLess
assertTrueall)selfr
   npr!   XYZbxbybzgxgygznthreadsr   r   host_arrr	   r   s                    @@r    test_ex_3d_gridTestRandom.test_ex_3d_grid
   s   	F		V 
	V  a 

 7R<"$r)B. 1B
 q	<r,,-c> ##%2778,d3)40x3/0x3/0r#    N)__name__
__module____qualname____firstlineno__r@   __static_attributes__rB   r#   r    r   r      s    -1r#   r   __main__)unittestnumba.cuda.testingr   r   r   rC   mainrB   r#   r    <module>rL      sH     < GH.1 .1 I.1b zMMO r#   