
    i                         S SK r S SKrS SKrS SKJr  S SKJrJ	r	  S SK
r
\" \S5      r\ R                  S:H  rS r\" S5       " S S	\	5      5       r\S
:X  a  \
R$                  " 5         gg)    N)cuda)skip_on_cudasimCUDATestCaseget_contextposixc                     SSK Jn   [        R                  " [        R
                  " S5      5        U R                  S 5        g ! U a  nU R                  U5         S nAg S nAff = f)Nr   )CudaDriverError   )numba.cuda.cudadrv.errorr	   r   	to_devicenparangeput)qr	   es      g/var/www/html/trading/venv/lib/python3.13/site-packages/numba/cuda/tests/cudapy/test_multiprocessing.py	fork_testr      sJ    8ryy|$ 	
d  	as   *A A%
A  A%zdisabled for cudasimc                   r    \ rS rSr\R
                  " \S5      \R
                  " \S5      S 5       5       rSr	g)TestMultiprocessing   zrequires mp.get_contextzrequires Unixc                 \   [         R                  " 5         [        R                  " S5      nUR	                  5       nUR                  [        U/S9nUR                  5         UR                  5       nUR                  5         U R                  U5        U R                  S[        U5      5        g)z
Test fork detection.
fork)targetargszCUDA initialized before forkingN)r   current_contextmpr   QueueProcessr   startgetjoinassertIsNotNoneassertInstr)selfctxr   procexcs        r   	test_forkTestMultiprocessing.test_fork   s{     	nnV$IIK{{)1#{6

eeg		S!7SB     N)
__name__
__module____qualname____firstlineno__unittest
skipUnlesshas_mp_get_contextis_unixr)   __static_attributes__r,   r+   r   r   r      s;    +-FG/2C 3 HCr+   r   __main__)osmultiprocessingr   numpyr   numbar   numba.cuda.testingr   r   r1   hasattrr3   namer4   r   r   r-   mainr,   r+   r   <module>r?      ss    	    < R/ 
''W
 '(C, C )C( zMMO r+   