
    wg                         d dl Z d dlmZ d dlZd dlmZ d dlmZm	Z	 d Z
 G d de	      Zedk(  r ej                          yy)	    N)Queue)cuda)unittestContextResettingTestCasec                 J   	 t        j                  d       t        j                         }t        j                  d      }t        j
                  ||      }|j                          ~~t        j                          y # t        $ r}| j                  |       Y d }~y d }~ww xY w)Nr   d   )stream)
r   select_devicer	   nparange	to_devicesynchronizeclose	Exceptionput)exception_queuer	   AdAes        p/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/cuda/tests/cudadrv/test_select_device.py	newthreadr      sy    
1IIcN^^Af-

 As   A;A> >	B"BB"c                       e Zd Zd Zy)TestSelectDevicec                 X   t               }t        d      D ]>  }t        j                  t        |f      }|j                          |j                          @ g }|j                         s0|j                  |j                                |j                         s0| j                  |g        y )N
   )targetargs)r   range	threadingThreadr   startjoinemptyappendgetassertEqual)selfr   it
exceptionss        r   test_select_devicez#TestSelectDevice.test_select_device   s    'r 	A  	8JKAGGIFFH	
 
!'')o1134 "'')R(    N)__name__
__module____qualname__r+    r,   r   r   r      s    
)r,   r   __main__)r   queuer   numpyr   numbar   numba.cuda.testingr   r   r   r   r-   mainr0   r,   r   <module>r7      sB        A)/ ) zHMMO r,   