
    wg1                         d dl Z d dlmZ d dlmZ d dlmZmZ d dlm	Z	  G d de      Z
edk(  r ej                          yy)	    N)cuda)driver)unittestContextResettingTestCase)Queuec                       e Zd Zd Zy)TestResetDevicec                 6   d }t               }t        j                  ||f      }|j                          |j	                          g }|j                         s0|j                  |j                                |j                         s0| j                  |g        y )Nc                    	 t        t        j                               }t        d      D ]2  }|D ]+  }t        j                  |       t        j
                          - 4 y # t        $ r}| j                  |       Y d }~y d }~ww xY w)N   )ranger   get_device_countr   select_deviceclose	Exceptionput)exception_queuedevices_des        o/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/cuda/tests/cudadrv/test_reset_device.py	newthreadz4TestResetDevice.test_reset_device.<locals>.newthread   sv    ' 7 7 9:q %A$ %**1-

%%  '##A&&'s   AA   	B)A??B)targetargs)	r   	threadingThreadstartjoinemptyappendgetassertEqual)selfr   r   t
exceptionss        r   test_reset_devicez!TestResetDevice.test_reset_device	   s}    	'  'I_4FG			
!'')o1134 "'')R(    N)__name__
__module____qualname__r'    r(   r   r	   r	      s    )r(   r	   __main__)r   numbar   numba.cuda.cudadrv.driverr   numba.cuda.testingr   r   queuer   r	   r)   mainr,   r(   r   <module>r3      s;      , A ). )6 zHMMO r(   