
    wg                     v    d dl Z d dlmZmZ  ed       G d de             Zedk(  r e j                          yy)    N)CUDATestCaseskip_on_cudasimz4cudasim doesn't support cuda import at non-top-levelc                       e Zd Zd Zy)
TestRandomc                 H   ddl m ddlm}m dd l}j                  fd       }d\  }}}d\  }}}	d\  }
}}||z  |	z  |
z  |z  |z  } ||d	      }j                  |||f|j                  
      } ||
||f|||	ff   ||       |j                         }| j                  |j                  |      d       | j                  |j                  |      d       | j                  |j                  |dk               | j                  |j                  |dk\               y )Nr   )cuda)create_xoroshiro128p_statesxoroshiro128p_uniform_float32c                 T   j                  d      \  }}}j                  d      \  }}}||z  |z  ||z  z   |z   }t        || j                  d   |      D ]O  }	t        || j                  d   |      D ]0  }
t        || j                  d   |      D ]  } ||      | |	|
|f<    2 Q y )N   r         )gridgridsizerangeshape)arr
rng_statesstartxstartystartzstridexstrideystrideztidijkr   r
   s               n/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/cuda/tests/doc_examples/test_random.py	random_3dz-TestRandom.test_ex_3d_grid.<locals>.random_3d   s     &*YYq\"FFF(,a(8%GWg G#g-&72BCfLC 6399Q<9 Vvsyy|W= VA"6399Q<A V'DZQT'UAq!GVVV    )i  i  i  )   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_gridzTestRandom.test_ex_3d_grid
   s-   	F		V 
	V  1a 
B
B 7R<"$r)B. 1B
 Aq	<-	2r2,R,-c:> ##%2778,d3)40x3/0x3/0r!   N)__name__
__module____qualname__r?    r!   r   r   r      s    -1r!   r   __main__)unittestnumba.cuda.testingr   r   r   r@   mainrC   r!   r   <module>rH      sJ     < GH.1 .1 I.1b zHMMO r!   